search

ultralytics / yolov5

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

Two targets combined to one by yolov5, which is not expected #1126

Closed universewill closed 3 years ago

universewill commented 3 years ago

Two targets animals combined to one by yolov5, which is not expected

glenn-jocher commented 3 years ago

larger model, larger image size, more training data, more epochs

universewill commented 3 years ago

larger model, larger image size, more training data, more epochs

Is editing .yaml file enough to change into a larger model? @glenn-jocher

glenn-jocher commented 3 years ago

@universewill you can experiment with yaml modifications if you are an expert user, but the easiest is simply to select a larger model, up to yolov5x. If you specify --weights you don't need to specify a --cfg:

python train.py --data custom.yaml --weights yolov5x.pt

github-actions[bot] commented 3 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.