فهرست منبع

Update extract_ms_patches.py

Lin Manhui 2 سال پیش
والد
کامیت
d8f117cde9
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  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):