yolov5/data/scripts/download_weights.sh

21 lines
523 B
Bash
Raw Normal View History

2020-05-30 08:04:54 +08:00
#!/bin/bash
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
2020-11-10 22:04:14 +08:00
# Download latest models from https://github.com/ultralytics/yolov5/releases
2021-07-28 05:23:41 +08:00
# Example usage: bash path/to/download_weights.sh
# parent
# └── yolov5
# ├── yolov5s.pt ← downloads here
# ├── yolov5m.pt
# └── ...
2020-05-30 08:04:54 +08:00
2020-11-10 21:28:45 +08:00
python - <<EOF
from utils.downloads import attempt_download
2020-11-10 21:28:45 +08:00
models = ['n', 's', 'm', 'l', 'x']
models.extend([x + '6' for x in models]) # add P6 models
for x in models:
2020-11-10 21:28:45 +08:00
attempt_download(f'yolov5{x}.pt')
EOF