resize

app.py

@@ -67,8 +67,16 @@ def resize_image_by_height(image, new_height=500):
     resized_image = cv2.resize(image, (new_width, new_height))
     return resized_image
 
+def resize_image_by_width(image, new_width=500):
+    import cv2
+    height, width, _ = image.shape
+    aspect_ratio = height / width
+    new_height = int(new_width * aspect_ratio)
+    resized_image = cv2.resize(image, (new_width, new_height))
+    return resized_image
+
 def predict(image, text, ground_tokens=""):
-    image = resize_image_by_height(image)
+    image = resize_image_by_width(image)
     ground_tokens = None if ground_tokens.strip() == "" else ground_tokens.strip().split(";")
     result, _ = glip_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)
     fiber_result, _ = fiber_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)