提交 3d2eecb9 编写于 作者: F FlyingQianMM

fix bugs in seg_visualize

上级 2b047093
...@@ -66,7 +66,7 @@ def visualize_segmentation(image, result, weight=0.6, save_dir='./'): ...@@ -66,7 +66,7 @@ def visualize_segmentation(image, result, weight=0.6, save_dir='./'):
image_name = str(int(time.time())) + '.jpg' image_name = str(int(time.time())) + '.jpg'
else: else:
image_name = os.path.split(image)[-1] image_name = os.path.split(image)[-1]
image = cv2.imread(image) im = cv2.imread(image)
vis_result = cv2.addWeighted(im, weight, pseudo_img, 1 - weight, 0) vis_result = cv2.addWeighted(im, weight, pseudo_img, 1 - weight, 0)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册