Update utils.py

utils.py

@@ -90,7 +90,11 @@ class UnionFind:
         return labels
 
 def visualise_single_image_prediction(image_as_np_array, predictions, filename):
-    figure, subplot = plt.subplots(1, 1, figsize=(10, 10))
+    h, w = image_as_np_array.shape[:2]
+    if h > w:
+        figure, subplot = plt.subplots(1, 1, figsize=(10, 10 * h / w))
+    else:
+        figure, subplot = plt.subplots(1, 1, figsize=(10 * w / h, 10))
     subplot.imshow(image_as_np_array)
     plot_bboxes(subplot, predictions["panels"], color="green")
     plot_bboxes(subplot, predictions["texts"], color="red", add_index=True)