Browse Source

Update extract_ms_patches.py

Lin Manhui 2 years ago
parent
commit
d8f117cde9
1 changed files with 2 additions and 2 deletions
  1. 2 2
      tools/extract_ms_patches.py

+ 2 - 2
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):