ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
50.58k stars 16.31k forks source link

Do we need augmentation when using YOLOv5 as our benchmark? #11013

Closed yaoshanliang closed 1 year ago

yaoshanliang commented 1 year ago

Search before asking

Question

I have a question that when using YOLOv5 as the benchmark, do we use default hyperparameters or close all augmentations, like hsv, translate, mosaic?

Additional

No response

github-actions[bot] commented 1 year 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 ⭐!

glenn-jocher commented 11 months ago

@yaoshanliang yes, data augmentation can improve model performance, especially when training on limited or imbalanced datasets. YOLOv5 includes several built-in augmentations, such as HSV color space transformations, scaling, and mosaic data augmentation, to enhance model generalization. For further details on YOLOv5's default augmentations and configuration, please refer to the Ultralytics YOLOv5 documentation. If you have any additional questions, feel free to ask!