examples/opencv_known_objects.py

"""
OpenCV Example 1: Known Object Detection (RT-DETR)
"""
import cv2
from object_intelligence import ObjectDetector

def main():
    detector = ObjectDetector()
    detector.set_mode("known")

    camera = cv2.VideoCapture(0)
    print("[INFO] Press 'q' to quit...")

    while True:
        success, frame = camera.read()
        if not success:
            print("[WARN] Failed to read frame from webcam.")
            break

        annotated_frame, detections = detector.detect_and_draw(frame)

        cv2.imshow("Known Object Detection (RT-DETR)", annotated_frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

    camera.release()
    cv2.destroyAllWindows()

if __name__ == "__main__":
    main()