Принудительно CPU в одиночку YoloV3 OpenVINO - PullRequest
Новая
       44

Принудительно CPU в одиночку YoloV3 OpenVINO

0 голосов
/ 22 апреля 2020

https://github.com/PINTO0309/OpenVINO-YoloV3

Я хочу иметь возможность прочитать это как изображение. Это не работает.

Как мне справиться? 

model_xml = 'model/hoge.xml'
model_bin = 'model/hoge.bin'
net = IENetwork.from_ir(model=model_xml, weights=model_bin)

plugin = IEPlugin(device='CPU', plugin_dirs=None)
plugin.add_cpu_extension("libcpu_extension_avx2.so")
exec_net = plugin.load(network=net)

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

model_n = 1
model_c = 3
model_h = 416
model_w = 416

del net

frame = cv2.imread('foo.jpg')

_h,  _w,  _ = frame.shape[:3]
in_frame = cv2.resize(frame, (model_w, model_h))
in_frame = in_frame.transpose((2, 0, 1))  # Change data layout from HWC to CHW
in_frame = in_frame.reshape((model_n, model_c, model_h, model_w))

exec_net.start_async(request_id=0, inputs={input_blob: in_frame})
...