UtensilDetector / inference /detect_webcam.py
anjali2002's picture
Complete Code
03efa53
raw
history blame
442 Bytes
from ultralytics import YOLO
import cv2
model = YOLO("runs/detect/weights/best.pt")
cap = cv2.VideoCapture(0) # 0 = default webcam
while cap.isOpened():
ret, frame = cap.read()
if not ret: break
results = model(frame, conf=0.5, imgsz=640, augment=False)
annotated_frame = results[0].plot()
cv2.imshow("Webcam Detection", annotated_frame)
if cv2.waitKey(1) == ord('q'): break
cap.release()
cv2.destroyAllWindows()