Cara yang benar untuk mendapatkan tensor output untuk model yang memiliki lebih dari satu output
- Punya tensor output untuk model kustom yang memiliki tiga output:
dari openvino.runtime impor Core
inti = Inti()
model = core.read_model(model="model.xml")
compiled_model = core.compile_model(model, "CPU")
infer_request = compiled_model.create_infer_request()
infer_request.start_async()
infer_request.menunggu()
keluaran = infer_request.get_output_tensor()
cetak (keluaran) - Kesalahan yang diterima:
RuntimeError: get_output_tensor() harus dipanggil pada fungsi dengan tepat satu parameter.
Metode ov:: InferRequest::get_output_tensor tanpa argumen dapat digunakan untuk model dengan hanya satu output.
- Gunakan metode ov:: InferRequest ::get_output_tensor dengan argumen (index: int) untuk model yang memiliki lebih dari satu output.
output1 = infer_request.get_output_tensor(0)
output2 = infer_request.get_output_tensor(1)
output3 = infer_request.get_output_tensor(2) - Gunakan atribut data objek Tensor untuk mengakses data tensor output untuk hasil inferensi.
output_buffer1 = output2.data
output_buffer2 = output2.data
output_buffer3 = output3.data
print(output_buffer1)
print(output_buffer2)
print(output_buffer3)