darknet parse all net to be tested

This commit is contained in:
Francesco Gatti
2020-06-01 12:22:55 +02:00
parent d8fbee58d8
commit d2e2669b6d
74 changed files with 560 additions and 4940 deletions
+2 -1
View File
@@ -59,7 +59,8 @@ Network::Network(dataDim_t input_dim) {
}
Network::~Network() {
for(int i=0; i<num_layers; i++)
delete layers[i];
checkCUDNN( cudnnDestroy(cudnnHandle) );
checkERROR( cublasDestroy(cublasHandle) );
}
+1 -1
View File
@@ -122,7 +122,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
input = Ilay->getOutput(0);
input->setName( (l->getLayerName() + std::to_string(i) + "_out").c_str() );
if(l->getLayerType() == LAYER_YOLO || l->final)
if(l->final)
networkRT->markOutput(*input);
tensors[l] = input;
}
+2 -1
View File
@@ -13,7 +13,8 @@ namespace tk { namespace dnn {
Yolo::Yolo(Network *net, int classes, int num, std::string fname_weights, int n_masks, float scale_xy) :
Layer(net) {
this->final = true;
this->classes = classes;
this->num = num;
this->n_masks = n_masks;