glenn-jocher commited on
Commit
27138c2
·
unverified ·
1 Parent(s): 67bf9a9

Update download_weights.sh (#1338)

Browse files
Files changed (1) hide show
  1. weights/download_weights.sh +8 -8
weights/download_weights.sh CHANGED
@@ -1,10 +1,10 @@
1
  #!/bin/bash
2
- # Download common models
3
 
4
- python -c "
5
- from utils.google_utils import *;
6
- attempt_download('weights/yolov5s.pt');
7
- attempt_download('weights/yolov5m.pt');
8
- attempt_download('weights/yolov5l.pt');
9
- attempt_download('weights/yolov5x.pt')
10
- "
 
1
  #!/bin/bash
2
+ # Download latest model from https://github.com/ultralytics/yolov5/releases
3
 
4
+ python - <<EOF
5
+ from utils.google_utils import attempt_download
6
+
7
+ for x in ['s', 'm', 'l', 'x']:
8
+ attempt_download(f'yolov5{x}.pt')
9
+
10
+ EOF