Update extract_ms_patches.py

2 years ago · d8f117cde9
parent dc60b44b49
commit d8f117cde9
1 changed files with 2 additions and 2 deletions
--- a/tools/extract_ms_patches.py
+++ b/tools/extract_ms_patches.py
@ -193,8 +193,8 @@ class QuadTree(object):
        vis = np.ascontiguousarray(im)
        for node in nodes:
            i, j, h, w = node.coords
-            vis = cv2.rectangle(vis, (j, i), (j + w, i + h), (0, 0, 255), 2)
-        cv2.imwrite(save_path, vis)
+            vis = cv2.rectangle(vis, (j, i), (j + w, i + h), (255, 0, 0), 2)
+        cv2.imwrite(save_path, vis[...,::-1])
        return save_path

    def print_tree(self, node=None, level=0):