Update utils.py

utils.py

@@ -3,6 +3,8 @@ import torch
 import numpy as np
 import cv2
 import random
+import matplotlib.patches as patches
+import config
 
 from pytorch_grad_cam.base_cam import BaseCAM
 from pytorch_grad_cam.utils.svd_on_activations import get_2d_projection
@@ -181,4 +183,23 @@ def draw_bounding_boxes(image, boxes, class_labels):
             lineType=cv2.LINE_AA,
         )
 
-    return image
+    return image
+
+def get_annotations(nms_boxes_output,width, height):
+    for box in nms_boxes_output[0]:
+        class_prediction = int(box[0])
+        box = box[2:]
+    
+        upper_left_x = box[0] - box[2] / 2
+        upper_left_y = box[1] - box[3] / 2
+        rect = patches.Rectangle(
+                (upper_left_x * width, upper_left_y * height),
+                box[2] * width,
+                box[3] * height,
+                linewidth=2,
+                edgecolor=colors[class_prediction],
+                facecolor="none",
+        )
+        rect = rect.get_bbox().get_points()
+        annotations.append([rect[0].astype(int).tolist()+rect[1].astype(int).tolist(),
+                            config.PASCAL_CLASSES[class_prediction]])