ultralytics / yolov5

YOLOv5 πŸš€ in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
48.73k stars 15.93k forks source link

varifocal loss #5339

Closed zwhua006 closed 2 years ago

zwhua006 commented 2 years ago

❔Question

Additional context

Recently I read the paper and I want to know if you have tried varifocal loss instead of BCE loss, I guess it will works.

glenn-jocher commented 2 years ago

@zwhua006 hi, thank you for your feature suggestion on how to improve YOLOv5 πŸš€! No, I have not tried varifocal loss with YOLOv5. It sounds like an interesting idea though!

The fastest and easiest way to incorporate your ideas into the official codebase is to submit a Pull Request (PR) implementing your idea, and if applicable providing before and after profiling/inference/training results to help us understand the improvement your feature provides. This allows us to directly see the changes in the code and to understand how they affect workflows and performance.

Please see our βœ… Contributing Guide to get started.

github-actions[bot] commented 2 years ago

πŸ‘‹ Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 πŸš€ resources:

Access additional Ultralytics ⚑ resources:

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 πŸš€ and Vision AI ⭐!