Bug in Metrics.py: Division by 0

[UnglvKitDe]

Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Training, Validation

Bug

@glenn-jocher Maybe similar to #8539, divisions with 0 may occur in methods box_iou(...) and wh_iou(...).

yolov5/utils/metrics.py

Lines 279 to 280 in a84cd02

	# IoU = inter / (area1 + area2 - inter)
	return inter / (box_area(box1.T)[:, None] + box_area(box2.T) - inter)

yolov5/utils/metrics.py

Line 310 in a84cd02

return inter / (wh1.prod(2) + wh2.prod(2) - inter) # iou = inter / (area1 + area2 - inter)

Environment

YOLOv5 torch 1.11 (cuda 11.3) and 1.12 (cuda 11.6)

Minimal Reproducible Example

By using torch.zeros(...) for both arguments (extreme case)

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[glenn-jocher]

@UnglvKitDe good news 😃! Your original issue may now be fixed ✅ in PR #8550. To receive this update:

• Git – git pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
• PyTorch Hub – Force-reload model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
• Notebooks – View updated notebooks
• Docker – sudo docker pull ultralytics/yolov5:latest to update your image

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!

[UnglvKitDe]

@glenn-jocher Thank you :)

[glenn-jocher]

@UnglvKitDe you're welcome! 😊 I'm glad I could help. If you have any more questions or need further assistance, feel free to reach out. Good luck with your YOLOv5 projects!