浏览代码

fix bugs in seg_visualize

FlyingQianMM 5 年之前
父节点
当前提交
3d2eecb9ee
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/models/utils/visualize.py

+ 1 - 1
paddlex/cv/models/utils/visualize.py

@@ -66,7 +66,7 @@ def visualize_segmentation(image, result, weight=0.6, save_dir='./'):
         image_name = str(int(time.time())) + '.jpg'
     else:
         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)