|
@@ -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):
|