zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Weights download script minor improvements (#5213) 

* Add nano model to the download list

* Add possibility to also download the "*6" models

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Cristi Fati 2021-10-23 15:20:26 +03:00 committed by GitHub
parent e0c3f42de7
commit fee83c1634
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
data/scripts/download_weights.sh
View File

@ -11,7 +11,10 @@
python - <<EOF python - <<EOF
from utils.downloads import attempt_download from utils.downloads import attempt_download
for x in ['s', 'm', 'l', 'x']: 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') attempt_download(f'yolov5{x}.pt')
EOF EOF