resize

app.py

@@ -59,29 +59,11 @@ athetics_params = {
     "text_offset_original": 8, # distance between text and box
 }
 
-def resize_image_by_height(image, new_height=500):
-    import cv2
-    height, width, _ = image.shape
-    aspect_ratio = width / height
-    new_width = int(new_height * aspect_ratio)
-    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_width(image)
-    print(image.shape)
-    max_len = max(image.shape)
+    img_len = min(image.shape[:2])
     import math
-    athetics_params["text_size"] = math.ceil(max_len/1000)
-    athetics_params["text_pixel"] = math.ceil(max_len/1000*3)
+    athetics_params["text_size"] = math.ceil(img_len/1000)
+    athetics_params["text_pixel"] = math.ceil(img_len/1000*3)
     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)