test-model.py

Go to the documentation of this file.

#!/usr/bin/python3
 
import cv2
# Parameters
MODEL_FPATH = "face-detection-retail-0004.bin"
ARCH_FPATH = "face-detection-retail-0004.xml"
CONF_THRESH = 0.5 # confidence of each object detected
 
# Load the model.
net = cv2.dnn.readNet(ARCH_FPATH, MODEL_FPATH)
 
# Specify NCS2 Myriad as a target device.
net.setPreferableBackend(cv2.dnn.DNN_BACKEND_INFERENCE_ENGINE)
net.setPreferableTarget(cv2.dnn.DNN_TARGET_MYRIAD) # DNN_TARGET_CPU (only if you don't have NCS2) or DNN_TARGET_MYRIAD
 
vid_cap = cv2.VideoCapture(0)
if not vid_cap.isOpened():
    raise IOError("Webcam cannot be opened!")
 
while True:
    # Capture frames
    ret, frame = vid_cap.read()
    
    # Prepare input blob and perform inference
    blob = cv2.dnn.blobFromImage(frame, size=(300, 300), ddepth=cv2.CV_8U)
    net.setInput(blob)
    out = net.forward()
    
    # Draw detected faces
    for detect in out.reshape(-1, 7):
        conf = float(detect[2])
        xmin = int(detect[3] * frame.shape[1])
        ymin = int(detect[4] * frame.shape[0])
        xmax = int(detect[5] * frame.shape[1])
        ymax = int(detect[6] * frame.shape[0])
 
        if conf > CONF_THRESH:
            cv2.rectangle(frame, (xmin, ymin), (xmax, ymax), color=(0, 255, 0), thickness=2)
    
    cv2.imshow('Input', frame)
    
    # Press "ESC" key to stop webcam
    if cv2.waitKey(1) == 27:
        break
 
# Release video capture object and close the window
vid_cap.release()
cv2.destroyAllWindows()
cv2.waitKey(1)