[update] add scale checking

Author: NatLee

main.py

@@ -4,14 +4,14 @@
 
 if __name__ == '__main__':
 
-    # Load config
+    # load config
     opt = Option()
     img_paths = opt.img_paths
 
-    # Pick first one
+    # pick first one
     raw_img = cv2.imread(opt.get_first_one().absolute().as_posix())
 
-    # Choose how to crop
+    # choose how to crop
     cropper = Cropper(raw_img)
 
     if cropper.cropped:
@@ -23,7 +23,7 @@
 
         last_size = None
 
-        # List all images to crop
+        # list all images to crop
         for img_path in img_paths:
 
             img = cv2.imread(img_path.absolute().as_posix())
@@ -54,6 +54,6 @@
             else:
                 last_size = size
 
-            # Cropping other images
+            # cropping other images
             cropped_img = img[starty:endy, startx:endx, :]
             cv2.imwrite((opt.output_path / img_path.name).absolute().as_posix(), cropped_img)

utils/cropper.py

@@ -1,23 +1,21 @@
-
 from tkinter import Tk
-
 import numpy as np
-
 import cv2
 
-
 class Cropper:
 
     def __init__(self, img) -> None:
 
-        self.ratio = 0.7
-
-        self.screen_width, self.screen_height = self.get_screen_info()
         self.img = img
+        self.screen_width, self.screen_height = self.get_screen_info()
 
         height, width, _ = self.img.shape
+        width_scale = width / self.screen_width
+        height_scale = height / self.screen_height
 
-        if width / self.screen_width > self.ratio or height / self.screen_height > self.ratio:
+        # check if image is too large, we need to resize it
+        if width_scale > 1 or height_scale > 1:
+            self.ratio = 1 / max(width_scale, height_scale)
             self.img = cv2.resize(self.img, None, fx=self.ratio, fy=self.ratio)
         else:
             self.ratio = 1.0