火焰面Openvino

时间:2020-07-17 09:08:11

标签: tensorflow model computer-vision regression openvino

嗨,我正在尝试使用openvino从blazeface模型创建地标。模型正在运行,我在回归输出中输出了一个形状数组(1,896,16),这些值对应于张量:


[10.116998 2.7768276 29.180841 29.190268 -7.6588407 -1.8769206 7.955852 2.8945901 -1.1223608 11.560151 4.533435 13.420406 -7.406376 -9.7577505 25.858068 -1.905251]

[10.240653 2.762262 29.685017 29.694557 -7.6490545 -2.103835 8.018975 2.6713266 -1.0454882 11.4031725 4.6832647 13.421963 -7.434411 -9.877215 26.122345 -2.0081995]

[10.195236 2.7623253 29.391754 29.400812 -7.406289 -2.2815979 7.871913 2.3147857 -1.0604903 10.664557 4.600431 12.801059 -7.033283 -9.625556 25.726923 -1.9212234]

[9.646918 2.7344685 28.515514 28.524525 -7.4333186 -1.8820286 7.6982303 2.9067798 -1.1013308 11.309927 4.33471 13.137727 -7.308227 -9.689351 24.914877 -1.7770973]

问题是我不知道blazeface模型输出给我什么样的价值以及如何从中创建地标。我看到他们在回归导致tensorflow后使用了get_tensor函数,但是我没有找到与openvino相同的函数:

https://gist.github.com/ogl4jo3/5ba673b61ecc82780299edbcdcca301a

如果有人有想法,请

1 个答案:

答案 0 :(得分:0)

我简要查看了您提供的示例,发现get_tensor用于从网络获取原始结果。要获取盒子,您需要使用ProcessCPUorig_nms

我建议您使用OpenVINO的Inference Engine的Python API。 用法示例为the link

您可以尝试使上述示例适合您的用例,并使用功能ProcessCPUorig_nms从原始数据中提取框。