darknet parser tested

This commit is contained in:
Francesco Gatti
2020-06-01 14:55:08 +02:00
parent 21518c5eef
commit af9bb411be
21 changed files with 112 additions and 30 deletions
+6 -2
View File
@@ -59,12 +59,16 @@ 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) );
}
void Network::releaseLayers() {
for(int i=0; i<num_layers; i++)
delete layers[i];
num_layers = 0;
}
dnnType* Network::infer(dataDim_t &dim, dnnType* data) {
//do infer for every layer
+1 -2
View File
@@ -12,8 +12,7 @@
namespace tk { namespace dnn {
Region::Region(Network *net, int classes, int coords, int num) :
Layer(net) {
Layer(net) {
this->classes = classes;
this->coords = coords;
this->num = num;