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"};