Update download_weights.sh (#1338)
This commit is contained in:
1 parent
67bf9a974e
commit
27138c2ebb
1 file changed
+8
-8
@@ -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
|
||||||
Reference in new issue
Block a user