Update detection.py

detection.py

@@ -44,19 +44,21 @@ class ObjectDetection:
         rgb = ImageColor.getcolor(code, "RGB")
         return rgb 
     
-    def plot_bboxes(self, results, frame, threshold=0.5, box_color='red', text_color='white'):
+    def plot_bboxes(self, results, frame, threshold=0.5, box_color='orange', text_color='black'):
         labels, conf, coord = results
-
-        frame = frame.copy()
+    
+        # Resize the frame to 400x400
+        frame = cv2.resize(frame, (400, 400))
+    
         box_color = self.get_color(box_color)
         text_color = self.get_color(text_color)
-
+    
         for i in range(len(labels)):
             if conf[i] >= threshold:
                 x1, y1, x2, y2 = self.get_coords(frame, coord[i])
                 class_name = self.class_to_label(labels[i])
-
+    
                 cv2.rectangle(frame, (x1, y1), (x2, y2), box_color, 2)
                 cv2.putText(frame, f"{class_name} - {conf[i]*100:.2f}%", (x1, y1), cv2.FONT_HERSHEY_COMPLEX, 0.5, text_color)
-        
-        return frame
+    
+        return frame