From fda12f766e89079e38eaef7325b66e1d7f9c1304 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=BB=91cyc=E5=AE=87?= <1251519599@qq.com> Date: Thu, 5 Jan 2023 10:31:03 +0000 Subject: [PATCH] update ACL_TensorFlow/contrib/cv/Yolov5_for_ACL/offline_inference/postprocess.py. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 黑cyc宇 <1251519599@qq.com> --- .../contrib/cv/Yolov5_for_ACL/offline_inference/postprocess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ACL_TensorFlow/contrib/cv/Yolov5_for_ACL/offline_inference/postprocess.py b/ACL_TensorFlow/contrib/cv/Yolov5_for_ACL/offline_inference/postprocess.py index cd06a8663..91772dbe0 100644 --- a/ACL_TensorFlow/contrib/cv/Yolov5_for_ACL/offline_inference/postprocess.py +++ b/ACL_TensorFlow/contrib/cv/Yolov5_for_ACL/offline_inference/postprocess.py @@ -92,7 +92,7 @@ class COCOevaluator: test_data, _, batch_image_id, scale, dw, dh = fetcher.process_annotations(annotations) pred_result = [] for j in range(6): - output_data = np.fromfile(inference_path + '/davinci_' + str(k).zfill(6) + '_output' + str(j) + '.bin', + output_data = np.fromfile(inference_path + '/davinci_' + str(batch_image_id).zfill(12) + '_output' + str(j) + '.bin', dtype='float32') pred_result.append(output_data) k += 1 -- Gitee