diff --git a/scripts/download_weights.ps1 b/scripts/download_weights.ps1 new file mode 100755 index 00000000..bc352606 --- /dev/null +++ b/scripts/download_weights.ps1 @@ -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" diff --git a/scripts/setup.ps1 b/scripts/setup.ps1 index acc29458..b846bab1 100755 --- a/scripts/setup.ps1 +++ b/scripts/setup.ps1 @@ -22,7 +22,7 @@ else { $features = "full" } else{ - $features = "opencv-base,weights,weights-train" + $features = "opencv-base" } } diff --git a/scripts/setup.sh b/scripts/setup.sh index f0970907..12a769b8 100755 --- a/scripts/setup.sh +++ b/scripts/setup.sh @@ -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