From 5900c4389dc0d25ddb621c841f493416fcf3dca2 Mon Sep 17 00:00:00 2001
From: Erick <ahmederick00@gmail.com>
Date: Thu, 30 Jun 2022 17:33:52 +0200
Subject: [PATCH] Create image resizer script

---
 src/img-resizer/img-resizer.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 src/img-resizer/img-resizer.py

diff --git a/src/img-resizer/img-resizer.py b/src/img-resizer/img-resizer.py
new file mode 100644
index 0000000..27911b1
--- /dev/null
+++ b/src/img-resizer/img-resizer.py
@@ -0,0 +1,24 @@
+import numpy as np
+import cv2
+import os
+from matplotlib import pyplot as plt
+
+inputFolder  = 'dataset/dataset-original'
+outputFolder = '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")
+
+for filename in os.listdir(inputFolder):
+    if filename.endswith(".jpg") or filename.endswith(".jpeg"):
+        img_org = cv2.imread(filename)
+        img_res = cv2.resize(img_org, resolution)
+        cv2.imwrite(os.path.join(outputFolder, filename), img_res)
+    else:
+        continue
+
+print(cv2.__version__)
\ No newline at end of file