fix loading error in folder format

LiheYoung · Jan 22, 2024 · f76ccdd · f76ccdd
1 parent c95c9e7
commit f76ccdd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/run.py b/run.py
@@ -70,7 +70,8 @@
         filenames.sort()
 
     for filename in tqdm(filenames):
-        raw_image = cv2.imread(filename)
+        full_name = os.path.join(args.img_path, filename)
+        raw_image = cv2.imread(full_name)
         image = cv2.cvtColor(raw_image, cv2.COLOR_BGR2RGB) / 255.0
 
         h, w = image.shape[:2]