Fix download weights yolov3 versions

Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-03-14 18:20:32 +01:00
parent 555b32b5dc
commit 82d907971c
4 changed files with 7 additions and 6 deletions
@@ -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++) {