Update pipeline.py

pipeline.py

@@ -2,6 +2,7 @@ from ultralytics import YOLO
 from PIL import Image
 import numpy as np
 import cv2 as cv
+import pandas as pd
 
 
 class detectPipeline():
@@ -21,6 +22,7 @@ class detectPipeline():
         for sign in detections.boxes.data.tolist():
             x1, y1, x2, y2, score, class_id = sign
             sign_detections.append([int(x1), int(y1), int(x2), int(y2), score, int(class_id)])
+        print(sign_detections)
         return sign_detections
 
     def drawDetections2Image(self, img_path, detections):
@@ -33,3 +35,11 @@ class detectPipeline():
                         color=(0, 0, 255), lineType=cv.LINE_AA, thickness=10)
         img_detections = np.array(img)
         return img_detections
+
+    # get sign_detetction
+    def extractTextResults(self, detections):
+        text_results = ''
+        for bbox in detections:
+            x1, y1, x2, y2, score, class_id = bbox
+            text_results += f'{self.class_names[class_id]} : {score}\n'
+        return text_results