KB-VQA / models /yolov5 /data /scripts /download_weights.sh
m7mdal7aj's picture
uploading a fresh copy of the detection models
92894b3 verified
raw
history blame
641 Bytes
#!/bin/bash
# YOLOv5 πŸš€ by Ultralytics, AGPL-3.0 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