OpenVINO【车辆检测】

openvino简直就是一个外挂！！！

在我的i5-9300H 帧率高达90fps！！！ 只能用离谱二字来形容

# -*- coding: utf-8 -*-
"""
Created on Tue Nov 16 03:01:53 2021

@MysteriousKnight: 23608
@Email: xingchenziyi@163.com   

"""

from openvino.inference_engine import IECore
import numpy as np
import time
import cv2 as cv

def vehicle():
    ie = IECore()

    devices = ie.available_devices
    print(devices[:])

    # with open("imagenet_classes.txt") as f:
    #     labels = [line.strip() for line in f.readlines()]
        
    model_xml = "vehicle-detection-adas-0002.xml"
    model_bin = "vehicle-detection-adas-0002.bin"

    net = ie.read_network(model=model_xml, weights=model_bin)

    input_blob = next(iter(net.inputs))
    output_blob = next(iter(net.outputs))


    n,c,h,w = net.inputs[input_blob].shape
    cap = cv.VideoCapture(1)
    exec_net = ie.load_network(network=net, device_name="CPU")
    _, src = cap.read()
    while 1:
        _, src = cap.read()

        # src = cv.imread("C:/Users/23608/Desktop/xxx.jpg")
        img = cv.resize(src, (w,h))
        img = img.transpose(2,0,1)

        inf_start = time.time()
        res = exec_net.infer(inputs={input_blob:np.array([img])})
        inf_end = time.time() - inf_start
        ih,iw,ic = src.shape
        res = res[output_blob]
        for _ in res[0][0]:
            if _[2] > 0.5:
                index_ = _[1]
                xmin = int(_[3] * iw)
                ymin = int(_[4] * ih)
                xmax = int(_[5] * iw)
                ymax = int(_[6] * ih)
                cv.rectangle(src, (xmin,ymin), (xmax,ymax), (0,0,255), 2, 8)
                cv.putText(src, str(round(_[2],2)), (xmin,ymin - 10), cv.FONT_HERSHEY_SIMPLEX, 0.75, (255, 0, 0), 2)
        cv.putText(src, "infer time(ms): %.3f, FPS: %.2f"%(inf_end*1000, 1/(inf_end+0.0001)), (10, 50),
               cv.FONT_HERSHEY_SIMPLEX, 1.0, (255, 0, 255), 2, 8)
        cv.imshow("vehicle", src)
        if cv.waitKey(1) & 0xFF == ord('q'):
            break

    cv.waitKey(0)
    cv.destroyAllWindows()

if __name__ == "__main__":
    vehicle()

效果