正在尝试编写OpenVINO推理引擎，但将图像数据类型输入到FP16 get ValueError:无法将字符串转换为浮点型

Question

我用OpenVINO来尝试这个tutorial来创建我自己的推理引擎。当我尝试在inference_request中创建随机输入数据时，它可以正常工作。

random_input_data = np.random.randn(1, 3, 224, 224).astype(np.float16)
tensor_description = TensorDesc(precision="FP16", dims=(1, 3, 224, 224), layout='NCHW')
input_blobs = Blob(tensor_description, random_input_data)

但是如果我用我的真实图像路径替换random_input_data

random_input_data = np.array(r"C:\MyImagePath\XXX.png").astype(np.float16)

我收到错误消息

ValueError: could not convert string to float: 'C:\\MyImagePath\\XXX.png'

为什么我不能将我的真实镜像astype到FP16？(我使用带有openvino_2021.4.689的Windows10)

Answer 1

您可以使用cv2.imread("image.png")。

我建议您参考官方的OpenVINO文档：Integrate Inference Engine with Your Python Application

请记住，您需要准确了解模型的输入形状、布局和输入数据精度(FP32/FP16/等)才能获得正确的输出