Преглед изворни кода

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)
         vis = np.ascontiguousarray(im)
         for node in nodes:
         for node in nodes:
             i, j, h, w = node.coords
             i, j, h, w = node.coords
-            vis = cv2.rectangle(vis, (j, i), (j + w, i + h), (0, 0, 255), 2)
+            vis = cv2.rectangle(vis, (j, i), (j + w, i + h), (255, 0, 0), 2)
-        cv2.imwrite(save_path, vis)
+        cv2.imwrite(save_path, vis[...,::-1])
         return save_path
         return save_path
 
 
     def print_tree(self, node=None, level=0):
     def print_tree(self, node=None, level=0):