系列 | OpenVINO视觉加速库使用七

- CMake3.7以上版本
- VS2015/VS2017
- Python 3.6.5

%CV_SDK_DIR%\deployment_tools\inference_engine\samples\create_msvc2015_solution.bat

%CV_SDK_DIR%\deployment_tools\inference_engine\samples\create_msvc2017_solution.bat

%CV_SDK_DIR%\deployment_tools\inference_engine\bin\intel64\Release

%CV_SDK_DIR%\python\python3.6

C:\Users\Administrator\AppData\Local\Programs\Python\Python36\Lib\site-packages

from openvino.inference_engine import IENetwork, IEPlugin

# 加载MKLDNN - CPU Target
log.basicConfig(format="[ %(levelname)s ] %(message)s", level=log.INFO, stream=sys.stdout)
plugin = IEPlugin(device="CPU", plugin_dirs=plugin_dir)
plugin.add_cpu_extension(cpu_extension)

# 加载IR
log.info("Reading IR...")
net = IENetwork(model=model_xml, weights=model_bin)

# 获取输入输出层
input_blob = next(iter(net.inputs))
out_blob = next(iter(net.outputs))
log.info("Loading IR to the plugin...")

# 创建可执行网络
exec_net = plugin.load(network=net, num_requests=2)
# Read and pre-process input image
n, c, h, w = net.inputs[input_blob].shape
del net

# 开启同步或者异步执行模式
inf_start = time.time()
if is_async_mode:
    in_frame = cv2.resize(next_frame, (w, h))
    in_frame = in_frame.transpose((2, 0, 1))  # Change data layout from HWC to CHW
    in_frame = in_frame.reshape((n, c, h, w))
    exec_net.start_async(request_id=next_request_id, inputs={input_blob: in_frame})
else:
    in_frame = cv2.resize(frame, (w, h))
    in_frame = in_frame.transpose((2, 0, 1))  # Change data layout from HWC to CHW
    in_frame = in_frame.reshape((n, c, h, w))
    exec_net.start_async(request_id=cur_request_id, inputs={input_blob: in_frame})

# 解析DetectionOut
res = exec_net.requests[cur_request_id].outputs[out_blob]
for obj in res[0][0]:
    # Draw only objects when probability more than specified threshold
    if obj[2] > 0.5:
        xmin = int(obj[3] * initial_w)
        ymin = int(obj[4] * initial_h)
        xmax = int(obj[5] * initial_w)
        ymax = int(obj[6] * initial_h)
        class_id = int(obj[1])

        # Draw box and label\class_id
        cv2.rectangle(frame, (xmin, ymin), (xmax, ymax), (0, 0, 255), 2)
        cv2.putText(frame, "face" + ' ' + str(round(obj[2] * 100, 1)) + ' %', (xmin, ymin - 7),
                    cv2.FONT_HERSHEY_COMPLEX, 0.6, (255, 0, 0), 1)

# 显示绘制文本
# inf_time_message = "Inference time: N\A for async mode" if is_async_mode else \
inf_time_message = "Inference time: {:.3f} ms， FPS:{:.3f}".format(det_time * 1000, 1000 / (det_time * 1000))
render_time_message = "OpenCV rendering time: {:.3f} ms".format(render_time * 1000)
async_mode_message = "Async mode is on. Processing request {}".format(cur_request_id) if is_async_mode else \
    "Async mode is off. Processing request {}".format(cur_request_id)

cv2.putText(frame, inf_time_message, (15, 15), cv2.FONT_HERSHEY_COMPLEX, 0.5, (255, 255, 0), 1)
cv2.putText(frame, render_time_message, (15, 30), cv2.FONT_HERSHEY_COMPLEX, 0.5, (10, 10, 200), 1)
cv2.putText(frame, async_mode_message, (10, int(initial_h - 20)), cv2.FONT_HERSHEY_COMPLEX, 0.5,
            (10, 10, 200), 1)

cv2.imshow("OpenVINO-Python-SSD-Demo@57558865", frame)

# ready for next frame
if is_async_mode:
    cur_request_id, next_request_id = next_request_id, cur_request_id
    frame = next_frame