yolov5 / data /scripts /download_weights.sh
Cristi Fati
Weights download script minor improvements (#5213)
fee83c1 unverified
raw
history blame
523 Bytes
#!/bin/bash
# YOLOv5 πŸš€ by Ultralytics, GPL-3.0 license
# Download latest models from https://github.com/ultralytics/yolov5/releases
# Example usage: bash path/to/download_weights.sh
# parent
# └── yolov5
# β”œβ”€β”€ yolov5s.pt ← downloads here
# β”œβ”€β”€ yolov5m.pt
# └── ...
python - <<EOF
from utils.downloads import attempt_download
models = ['n', 's', 'm', 'l', 'x']
models.extend([x + '6' for x in models]) # add P6 models
for x in models:
attempt_download(f'yolov5{x}.pt')
EOF