Create object_detection.py

KOSASIH · Jun 2, 2024 · a3a1452 · a3a1452
1 parent 21758f8
commit a3a1452
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/cosmic_pi_network/cvlib/object_detection.py b/src/cosmic_pi_network/cvlib/object_detection.py
@@ -0,0 +1,9 @@
+import cv2
+import numpy as np
+
+class ObjectDetection:
+    def __init__(self, yolo_v4: YOLOv4):
+        self.yolo_v4 = yolo_v4
+
+    def detect_objects(self, image: np.ndarray) -> list:
+        return self.yolo_v4.detect(image)