detectVideo.py

import cv2

# img = cv2.imread("./test2.png")
video = cv2.VideoCapture(0)

classNames = []
classFile = "coco.name"

with open(classFile, 'rt') as f:
    classNames= f.read().rstrip("\n").split("\n")
#print(classNames)

weightPath = "model/frozen_inference_graph.pb"
configPath = "model/ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt"


net = cv2.dnn_DetectionModel(weightPath, configPath)
net.setInputSize(320,320)
net.setInputScale(1.0/ 127.5)
net.setInputMean((127.5, 127.5, 127.5))
net.setInputSwapRB(True)

while True:
    ret, frame = video.read()
    cv2.imshow("frame", frame)
    classIds, confs, bboxs = net.detect(frame, confThreshold=0.5)

    for classId, conf , bbox  in zip(classIds, confs, bboxs):
        try:
            cv2.rectangle(frame, bbox, color=(0,255,0))
            cv2.putText(frame,classNames[classId-1], (bbox[0]+10, bbox[1]-10), cv2.FONT_HERSHEY_COMPLEX, 0.8, (0,255,0),1)
        except:
            print("Object not in dataset...")
    if cv2.waitKey(1)==ord("q"):
        break
video.release()
cv2.destroyAllWindows()
cv2.waitKey(0)