Browse Source

Update download_weights.sh (#1338)

5.0
Glenn Jocher GitHub 4 years ago
parent
commit
27138c2ebb
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
  1. +8
    -8
      weights/download_weights.sh

+ 8
- 8
weights/download_weights.sh View File

@@ -1,10 +1,10 @@
#!/bin/bash
# Download common models
# Download latest model from https://github.com/ultralytics/yolov5/releases

python -c "
from utils.google_utils import *;
attempt_download('weights/yolov5s.pt');
attempt_download('weights/yolov5m.pt');
attempt_download('weights/yolov5l.pt');
attempt_download('weights/yolov5x.pt')
"
python - <<EOF
from utils.google_utils import attempt_download
for x in ['s', 'm', 'l', 'x']:
attempt_download(f'yolov5{x}.pt')
EOF

Loading…
Cancel
Save