download weights using a local script, do not rely on vcpkg

This commit is contained in:
Stefano Sinigardi 2021-04-14 15:19:06 +02:00
parent 013d5e676d
commit 0ee4ff36b2
3 changed files with 40 additions and 3 deletions

37
scripts/download_weights.ps1 Executable file
View File

@ -0,0 +1,37 @@
#!/usr/bin/env pwsh
$url = "https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov4-tiny.weights"
$url = "https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov4.weights"
$url = "https://drive.google.com/u/0/uc?id=18yYZWyKbo4XSDVyztmsEcF9B_6bxrhUY&export=download"
Invoke-WebRequest -Uri $url -OutFile "yolov3-tiny-prn.weights"
$url = "https://pjreddie.com/media/files/yolov3.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov3.weights"
$url = "https://pjreddie.com/media/files/yolov3-openimages.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov3-openimages.weights"
$url = "https://pjreddie.com/media/files/yolov2.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov2.weights"
$url = "https://pjreddie.com/media/files/yolov3-tiny.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov3-tiny.weights"
$url = "https://pjreddie.com/media/files/yolov2-tiny.weights"
Invoke-WebRequest -Uri $url -OutFile "yolov2-tiny.weights"
$url = "https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.conv.29"
Invoke-WebRequest -Uri $url -OutFile "yolov4-tiny.conv.29"
$url = "https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.conv.137"
Invoke-WebRequest -Uri $url -OutFile "yolov4.conv.137"
$url = "https://pjreddie.com/media/files/darknet53.conv.74"
Invoke-WebRequest -Uri $url -OutFile "darknet53.conv.74"
$url = "https://pjreddie.com/media/files/darknet19_448.conv.23"
Invoke-WebRequest -Uri $url -OutFile "darknet19_448.conv.23"

View File

@ -22,7 +22,7 @@ else {
$features = "full"
}
else{
$features = "opencv-base,weights,weights-train"
$features = "opencv-base"
}
}

View File

@ -55,14 +55,14 @@ if [ "$install_cuda" = true ] ; then
features="full"
else
echo "Unable to auto-install CUDA on this Linux OS"
features="opencv-base,weights,weights-train"
features="opencv-base"
fi
fi
else
if [[ -v CUDA_PATH ]]; then
features="full"
else
features="opencv-base,weights,weights-train"
features="opencv-base"
fi
fi