Update services/detection_service.py

services/detection_service.py

@@ -9,9 +9,15 @@ class DetectionService:
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.model.to(self.device)
         self.model.eval()
+        self.frame_counter = 0
+        self.frame_skip = 5  # Process every 5th frame for performance
 
     def detect_objects(self, image, confidence_threshold=0.9):
-        """Detect objects in an image."""
+        """Detect objects in an image, skipping frames for performance."""
+        self.frame_counter += 1
+        if self.frame_counter % self.frame_skip != 0:
+            return []  # Skip detection for this frame
+
         inputs = self.processor(images=image, return_tensors="pt").to(self.device)
         with torch.no_grad():
             outputs = self.model(**inputs)