benchmark_app: fix FP16 image type (#15393)

This commit is contained in:
Zlobin Vladimir 2023-01-30 18:31:12 +04:00 committed by GitHub
parent 85597f866c
commit a19d50d4d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -216,12 +216,12 @@ ov::Tensor get_image_tensor(const std::vector<std::string>& files,
inputInfo.first,
filenames_used);
} else if (type == ov::element::f16) {
return create_tensor_from_image<short>(files,
inputId,
batchSize,
inputInfo.second,
inputInfo.first,
filenames_used);
return create_tensor_from_image<ov::float16>(files,
inputId,
batchSize,
inputInfo.second,
inputInfo.first,
filenames_used);
} else if (type == ov::element::f64) {
return create_tensor_from_image<double>(files,
inputId,