update

app.py

@@ -28,6 +28,12 @@ def dict2namespace(config):
 
 def load_img (filename, norm=True,):
     img = np.array(Image.open(filename).convert("RGB"))
+    h, w = img.shape[:2]
+    
+    if w > 1920 or h > 1080:
+        new_h, new_w = h // 4, w // 4
+        img = np.array(Image.fromarray(img).resize((new_w, new_h), Image.BICUBIC))
+ 
     if norm:
         img = img / 255.
         img = img.astype(np.float32)