OpenVINO API 2.0 输出张量无法转换为图像

Question

这是的后续问题。

我尝试使用 YOLOv4 模型对 OpenVINO API 2.0 进行推理。

dcm_file = "1037973"
ds = dcmread(dcm_file, force=True)
ds.PixelRepresentation = 0
ds_arr = ds.pixel_array
        
core = ov.Core()
model = core.read_model(model="frozen_darknet_yolov4_model.xml")
model.reshape([ds_arr.shape[0], ds_arr.shape[1], ds_arr.shape[2], 3])
compiled_model = core.compile_model(model, "CPU")
infer_request = compiled_model.create_infer_request()
input_tensor = ov.Tensor(array=ds_arr, shared_memory=True)
#infer_request.set_input_tensor(input_tensor)
infer_request.start_async()
infer_request.wait()
output_tensor1 = infer_request.get_output_tensor(0)
output_tensor2 = infer_request.get_output_tensor(1)
output_tensor3 = infer_request.get_output_tensor(2)

之后，我想将 output_tensor 转换为图像。

我在 OpenVINO 文档上引用了 Single Image Super Resolution and Super Resolution with PaddleGAN，但没有成功。

我也尝试用Image.fromarray来转换

错误总是发生在下面。

AttributeError: 'openvino.pyopenvino.Tensor' object has no attribute xxxxx

如何正确处理openvino.pyopenvino.Tensor？

我的环境是Windows11，版本openvino_2022.1.0.643。

随时下载我的 inference files 需要的。

（ZIP 文件包含 3 个 IR 文件和 1 个输入文件。）

Answer 1

使用数据attribute of the Tensor object访问输出张量数据。

output_tensor1 = infer_request.get_output_tensor(0)
output_tensor2 = infer_request.get_output_tensor(1)
output_tensor3 = infer_request.get_output_tensor(2)
output_buffer1 = output_tensor1.data
output_buffer2 = output_tensor2.data
output_buffer3 = output_tensor3.data
print(output_buffer1)
print(output_buffer2)
print(output_buffer3)

OpenVINO API 2.0 输出张量无法转换为图像

OpenVINO API 2.0 output tensor cannot convert to image

python

openvino

yolov4