From 03b03ad1fbec410487b8b8bd90d555f84f8297d0 Mon Sep 17 00:00:00 2001 From: Erick Date: Thu, 30 Jun 2022 18:14:11 +0200 Subject: [PATCH] Fixed path reading bugs --- src/img-resizer/img-resizer.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/src/img-resizer/img-resizer.py b/src/img-resizer/img-resizer.py index 27911b1..fae8efb 100644 --- a/src/img-resizer/img-resizer.py +++ b/src/img-resizer/img-resizer.py @@ -3,22 +3,20 @@ import cv2 import os from matplotlib import pyplot as plt -inputFolder = 'dataset/dataset-original' -outputFolder = 'dataset/dataset-resized' +inputFolder = os.path.join(os.getcwd(), 'dataset', 'dataset-original') +outputFolder = os.path.join(os.getcwd(), 'dataset', 'dataset-resized') resolution = tuple((640,640)) -if os.path.exists(outputFolder) is True: - try: - os.mkdir(outputFolder) - except AssertionError: - print ("Output path does not exist") +if os.path.exists(outputFolder) is False: + os.mkdir(outputFolder) for filename in os.listdir(inputFolder): if filename.endswith(".jpg") or filename.endswith(".jpeg"): - img_org = cv2.imread(filename) + img_org = cv2.imread(inputFolder + '/' + filename) img_res = cv2.resize(img_org, resolution) cv2.imwrite(os.path.join(outputFolder, filename), img_res) else: continue +#debug string print(cv2.__version__) \ No newline at end of file