mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Fix download weights yolov3 versions

Browse Source 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
This commit is contained in:
Micaela Verucchi
2020-03-14 18:20:32 +01:00
parent 555b32b5dc
commit 82d907971c
4 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/yolo3/yolo3.cpp
+2 -1
View File
@@ -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++) {
tests/yolo3_berkeley/yolo3_berkeley.cpp
+2 -1
View File
@@ -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++) {
tests/yolo3_coco4/yolo3_coco4.cpp
+1 -2
View File
@@ -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;
tests/yolo3_flir/yolo3_flir.cpp
+2 -2
View File
@@ -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"};