Detection is accelerated by 7 percent (fused conv and batch_norm layers)

This commit is contained in:
AlexeyAB
2018-04-03 16:42:00 +03:00
parent ee8a941922
commit 1b2c70f82a
4 changed files with 40 additions and 1 deletions

View File

@ -148,7 +148,7 @@ void demo(char *cfgfile, char *weightfile, float thresh, float hier_thresh, int
load_weights(&net, weightfile);
}
set_batch_network(&net, 1);
fuse_conv_batchnorm(net);
srand(2222222);
if(filename){