Update download_weights.sh (#1338)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-11-10 14:28:45 +01:00
1 parent 67bf9a974e
commit 27138c2ebb
1 file changed
+8 -8
+8 -8
View File
@@ -1,10 +1,10 @@
#!/bin/bash #!/bin/bash
# Download common models # Download latest model from https://github.com/ultralytics/yolov5/releases
python -c " python - <<EOF
from utils.google_utils import *; from utils.google_utils import attempt_download
attempt_download('weights/yolov5s.pt');
attempt_download('weights/yolov5m.pt'); for x in ['s', 'm', 'l', 'x']:
attempt_download('weights/yolov5l.pt'); attempt_download(f'yolov5{x}.pt')
attempt_download('weights/yolov5x.pt')
" EOF