From 7ac736b22c1184376c87432da4760b90dc54092b Mon Sep 17 00:00:00 2001 From: yuedongli1 Date: Tue, 22 Aug 2023 15:33:14 +0800 Subject: [PATCH] fix yolov4 EvalWhileTrain bug --- mindyolo/models/heads/yolov4_head.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mindyolo/models/heads/yolov4_head.py b/mindyolo/models/heads/yolov4_head.py index afbcaf0e..06a46f0f 100644 --- a/mindyolo/models/heads/yolov4_head.py +++ b/mindyolo/models/heads/yolov4_head.py @@ -115,4 +115,7 @@ def construct(self, x): return prediction, box_xy, box_wh box_xy *= input_shape box_wh *= input_shape - return ops.concat((box_xy, box_wh, box_confidence, box_probs), -1) + return ops.concat((box_xy.astype(ms.float32), + box_wh.astype(ms.float32), + box_confidence.astype(ms.float32), + box_probs.astype(ms.float32)), -1)