This commit is contained in:
Glenn Jocher 2019-12-06 17:33:17 -08:00
parent 4988397458
commit f2d47c1256
1 changed files with 10 additions and 5 deletions

View File

@ -1,20 +1,25 @@
#!/bin/bash #!/bin/bash
# make '/weights' directory if it does not exist and cd into it # make '/weights' directory if it does not exist and cd into it
mkdir -p weights && cd weights # mkdir -p weights && cd weights
# copy darknet weight files, continue '-c' if partially downloaded # copy darknet weight files, continue '-c' if partially downloaded
wget -c https://pjreddie.com/media/files/yolov3.weights # wget -c https://pjreddie.com/media/files/yolov3.weights
wget -c https://pjreddie.com/media/files/yolov3-tiny.weights # wget -c https://pjreddie.com/media/files/yolov3-tiny.weights
wget -c https://pjreddie.com/media/files/yolov3-spp.weights # wget -c https://pjreddie.com/media/files/yolov3-spp.weights
# yolov3 pytorch weights # yolov3 pytorch weights
# download from Google Drive: https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI # download from Google Drive: https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI
# darknet53 weights (first 75 layers only) # darknet53 weights (first 75 layers only)
wget -c https://pjreddie.com/media/files/darknet53.conv.74 # wget -c https://pjreddie.com/media/files/darknet53.conv.74
# yolov3-tiny weights from darknet (first 16 layers only) # yolov3-tiny weights from darknet (first 16 layers only)
# ./darknet partial cfg/yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.conv.15 15 # ./darknet partial cfg/yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.conv.15 15
# mv yolov3-tiny.conv.15 ../ # mv yolov3-tiny.conv.15 ../
# new method
python3 -c "from models import *;
attempt_download('weights/yolov3.pt');
attempt_download('weights/yolov3-spp.pt')"