Merge pull request #244 from Pluventi/master

Update postprocessing.py
This commit is contained in:
Vladimir Mandic
2023-04-21 09:28:31 -04:00
committed by GitHub
+1 -1
View File
@@ -18,7 +18,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
if extras_mode == 1:
for img in image_folder:
image = Image.open(img)
image = Image.open(img.name)
image_data.append(image)
image_names.append(os.path.splitext(img.orig_name)[0])
elif extras_mode == 2: