From 82d907971c2cda7c3df715eb229f972da2ce56f3 Mon Sep 17 00:00:00 2001 From: Micaela Verucchi Date: Sat, 14 Mar 2020 18:20:32 +0100 Subject: [PATCH] Fix download weights yolov3 versions Signed-off-by: Micaela Verucchi --- tests/yolo3/yolo3.cpp | 3 ++- tests/yolo3_berkeley/yolo3_berkeley.cpp | 3 ++- tests/yolo3_coco4/yolo3_coco4.cpp | 3 +-- tests/yolo3_flir/yolo3_flir.cpp | 4 ++-- 4 files changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/yolo3/yolo3.cpp b/tests/yolo3/yolo3.cpp index b3b2f0f..dac61b6 100644 --- a/tests/yolo3/yolo3.cpp +++ b/tests/yolo3/yolo3.cpp @@ -10,11 +10,12 @@ int main() { // create yolo3 model std::string bin_path = "../tests/yolo3"; + downloadWeightsifDoNotExist("../tests/yolo3/layers/input.bin", bin_path, "https://cloud.hipert.unimore.it/s/jPXmHyptpLoNdNR/download"); int classes = 80; tk::dnn::Yolo *yolo [3]; #include "models/Yolo3.h" - downloadWeightsifDoNotExist(input_bin, bin_path, "https://cloud.hipert.unimore.it/s/jPXmHyptpLoNdNR/download"); + // fill classes names for(int i=0; i<3; i++) { diff --git a/tests/yolo3_berkeley/yolo3_berkeley.cpp b/tests/yolo3_berkeley/yolo3_berkeley.cpp index 4c62f8e..d9f6dc2 100644 --- a/tests/yolo3_berkeley/yolo3_berkeley.cpp +++ b/tests/yolo3_berkeley/yolo3_berkeley.cpp @@ -10,11 +10,12 @@ int main() { // create yolo3 model std::string bin_path = "../tests/yolo3_berkeley"; + downloadWeightsifDoNotExist("../tests/yolo3_berkeley/layers/input.bin", bin_path, "https://cloud.hipert.unimore.it/s/o5cHa4AjTKS64oD/download"); int classes = 10; tk::dnn::Yolo *yolo [3]; #include "models/Yolo3.h" - downloadWeightsifDoNotExist(input_bin, bin_path, "https://cloud.hipert.unimore.it/s/o5cHa4AjTKS64oD/download"); + // fill classes names for(int i=0; i<3; i++) { diff --git a/tests/yolo3_coco4/yolo3_coco4.cpp b/tests/yolo3_coco4/yolo3_coco4.cpp index 61274b3..72c983c 100644 --- a/tests/yolo3_coco4/yolo3_coco4.cpp +++ b/tests/yolo3_coco4/yolo3_coco4.cpp @@ -10,12 +10,11 @@ int main() { // create yolo3 model std::string bin_path = "../tests/yolo3_coco4"; + downloadWeightsifDoNotExist("../tests/yolo3_coco4/layers/input.bin", bin_path, "https://cloud.hipert.unimore.it/s/o27NDzSAartbyc4/download"); int classes = 4; tk::dnn::Yolo *yolo [3]; #include "models/Yolo3.h" - downloadWeightsifDoNotExist(input_bin, bin_path, "https://cloud.hipert.unimore.it/s/o27NDzSAartbyc4/download"); - // Load input dnnType *data; dnnType *input_h; diff --git a/tests/yolo3_flir/yolo3_flir.cpp b/tests/yolo3_flir/yolo3_flir.cpp index fcb2ce5..f284d31 100644 --- a/tests/yolo3_flir/yolo3_flir.cpp +++ b/tests/yolo3_flir/yolo3_flir.cpp @@ -11,12 +11,12 @@ int main() { // create yolo3 model std::string bin_path = "../tests/yolo3_flir"; + downloadWeightsifDoNotExist("../tests/yolo3_flir/layers/input.bin", bin_path, "https://cloud.hipert.unimore.it/s/62DECncmF6bMMiH/download"); + int classes = 3; tk::dnn::Yolo *yolo [3]; #include "models/Yolo3.h" - downloadWeightsifDoNotExist(input_bin, bin_path, "https://cloud.hipert.unimore.it/s/62DECncmF6bMMiH/download"); - // fill classes names for(int i=0; i<3; i++) { yolo[i]->classesNames = {"person", "bike", "car"};