e9ab205ef6
* Refactor code for speed and clarity * Auto-format by https://ultralytics.com/actions * Update README.md Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
24 lines
665 B
Bash
Executable File
24 lines
665 B
Bash
Executable File
#!/bin/bash
|
|
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
|
|
# Download latest models from https://github.com/ultralytics/yolov5/releases
|
|
# Example usage: bash data/scripts/download_weights.sh
|
|
# parent
|
|
# └── yolov5
|
|
# ├── yolov5s.pt ← downloads here
|
|
# ├── yolov5m.pt
|
|
# └── ...
|
|
|
|
python - <<EOF
|
|
from utils.downloads import attempt_download
|
|
|
|
p5 = list('nsmlx') # P5 models
|
|
p6 = [f'{x}6' for x in p5] # P6 models
|
|
cls = [f'{x}-cls' for x in p5] # classification models
|
|
seg = [f'{x}-seg' for x in p5] # classification models
|
|
|
|
for x in p5 + p6 + cls + seg:
|
|
attempt_download(f'weights/yolov5{x}.pt')
|
|
|
|
EOF
|