diff --git a/paddlex/cv/models/utils/visualize.py b/paddlex/cv/models/utils/visualize.py index b65936bcb93905c12bff100027caab846d564c32..6982bdf62993598ab8f0d42e09af2e303a7957bb 100644 --- a/paddlex/cv/models/utils/visualize.py +++ b/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)