tol/image-manipulation-tools
Archived
1
0
Fork 0
Code Activity

Fixed path reading bugs

Browse Source
This commit is contained in:
Erick
2022-06-30 18:14:11 +02:00
parent 5900c4389d
commit 03b03ad1fb
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/img-resizer/img-resizer.py
+5 -7
View File
@@ -3,22 +3,20 @@ import cv2
import os import os
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
inputFolder = 'dataset/dataset-original' inputFolder = os.path.join(os.getcwd(), 'dataset', 'dataset-original')
outputFolder = 'dataset/dataset-resized' outputFolder = os.path.join(os.getcwd(), 'dataset', 'dataset-resized')
resolution = tuple((640,640)) resolution = tuple((640,640))
if os.path.exists(outputFolder) is True: if os.path.exists(outputFolder) is False:
try:
os.mkdir(outputFolder) os.mkdir(outputFolder)
except AssertionError:
print ("Output path does not exist")
for filename in os.listdir(inputFolder): for filename in os.listdir(inputFolder):
if filename.endswith(".jpg") or filename.endswith(".jpeg"): 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) img_res = cv2.resize(img_org, resolution)
cv2.imwrite(os.path.join(outputFolder, filename), img_res) cv2.imwrite(os.path.join(outputFolder, filename), img_res)
else: else:
continue continue
#debug string
print(cv2.__version__) print(cv2.__version__)