Skip to content

Commit

Permalink
Merge pull request #189 from yuedongli1/master
Browse files Browse the repository at this point in the history
fix yolov4 EvalWhileTrain bug
  • Loading branch information
zhanghuiyao authored Aug 23, 2023
2 parents 07997fd + 7ac736b commit 803ee79
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion mindyolo/models/heads/yolov4_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

0 comments on commit 803ee79

Please sign in to comment.