树莓派和计算棒实现图形识别 RaspBerry Pi4 with OpenVino and Movidius

 

import cv2 as cv
import sys
import logging as log

# openvino_fd_myriad.py
filename ='./'

log.basicConfig(format="[ %(levelname)s ] %(message)s", level=log.INFO, stream=sys.stdout)

if len(sys.argv) > 1 :
    filename = filename + sys.argv[1]
else:
    filename = filename + 'Girls.jpg'
    
log.info("Image is " + filename)
# Load the model.
net = cv.dnn.readNet('face-detection-adas-0001.xml','face-detection-adas-0001.bin')

# Specify target device.

net.setPreferableTarget(cv.dnn.DNN_TARGET_MYRIAD)

log.info("Inference model is face-detection-adas-0001")
# Read an image.
frame = cv.imread(filename)

if frame is None:
    raise Exception('Image not found!')

log.info("Perform an inference " + filename)
# Prepare input blob and perform an inference.
blob = cv.dnn.blobFromImage(frame, size=(672, 384), ddepth=cv.CV_8U)
net.setInput(blob)
out = net.forward()

i = 0
# Draw detected faces on the frame.
for detection in out.reshape(-1, 7):
    confidence = float(detection[2])
    xmin = int(detection[3] * frame.shape[1])
    ymin = int(detection[4] * frame.shape[0])
    xmax = int(detection[5] * frame.shape[1])
    ymax = int(detection[6] * frame.shape[0])
    if confidence > 0.5:
        cv.rectangle(frame, (xmin, ymin), (xmax, ymax), color=(0, 255, 0), thickness=1)
        cv.rectangle(frame, (xmin-1, ymin-1),(xmin+70, ymin-10), color=(0, 255, 0), thickness= -1)
        i = i + 1
        cv.putText(frame, 'Human : ' + str(i) + ' ' + str(confidence) , (xmin, ymin -2), cv.FONT_HERSHEY_SIMPLEX, 0.3, (0,0,0), thickness=1)

# Save the frame to an image file.
cv.imwrite(filename + '_out.png', frame)
log.info("Save the frame to an image file as png : " + filename)