Browse Source

fix bugs in seg_visualize

FlyingQianMM 5 years ago
parent
commit
3d2eecb9ee
1 changed files with 1 additions and 1 deletions
  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)