nicer

app.py

@@ -61,10 +61,6 @@ athetics_params = {
 
 def predict(image, text, ground_tokens=""):
     ground_tokens = None if ground_tokens.strip() == "" else ground_tokens.strip().split(";")
-    aspect_ratio = image.width / image.height
-    new_height = 500
-    new_width = int(new_height * aspect_ratio)
-    image = image.resize((new_width, new_height))
     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)
     return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]]

maskrcnn_benchmark/engine/predictor_glip.py

@@ -147,6 +147,13 @@ class GLIPDemo(object):
         top_predictions = self._post_process(predictions, thresh)
 
         result = original_image.copy()
+        def resize_image_by_height(image, new_height=500):
+            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_img
+        result = resize_image_by_height(result)
         if self.show_mask_heatmaps:
             return self.create_mask_montage(result, top_predictions)