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

Question

这是的后续问题。 /a>。

我尝试使用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转换为图像。

我已经引用了 and 与Paddlegan的超级分辨率在OpenVino文档上，但徒劳无功。

而且我还尝试使用image.fromarray将其转换。

错误总是在下面发生。

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

如何处理openvino.pyopenvino.tensor properly？

我的环境是Windows 11，带有OpenVino_2022.1.0.643版本。

Answer 1

使用数据对象访问输出张量数据。

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)