1

inference/data/video_reader.py

@@ -35,7 +35,8 @@ class VideoReader_221128_TransColorization(Dataset):
         else:
             self.size_dir = size_dir
 
-        flag_reverse = args.getattr('reverse', False) if args is not None else False
+        # flag_reverse = args.getattr('reverse', False) if args is not None else False
+        flag_reverse = False
         self.frames = [img for img in sorted(os.listdir(self.image_dir), reverse=flag_reverse) if (img.endswith('.jpg') or img.endswith('.png')) and not img.startswith('.')]
         self.palette = Image.open(path.join(mask_dir, sorted([msk for msk in os.listdir(mask_dir) if not msk.startswith('.')])[0])).getpalette()
         self.first_gt_path = path.join(self.mask_dir, sorted([msk for msk in os.listdir(self.mask_dir) if not msk.startswith('.')])[0])