mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Added include/darknet.h

Browse Source
This commit is contained in:
AlexeyAB
2019-01-06 23:51:38 +03:00
parent c56931dd75
commit 3ff5084590
30 changed files with 1823 additions and 308 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/parser.c
View File

@ -1249,3 +1249,28 @@ void load_weights(network *net, char *filename)
load_weights_upto(net, filename, net->n);
}
// load network & force - set batch size
network *load_network_custom(char *cfg, char *weights, int clear, int batch)
{
printf(" Try to load cfg: %s, weights: %s, clear = %d \n", cfg, weights, clear);
network *net = calloc(1, sizeof(network));
*net = parse_network_cfg_custom(cfg, batch);
if (weights && weights[0] != 0) {
load_weights(net, weights);
}
if (clear) (*net->seen) = 0;
return net;
}
// load network & get batch size from cfg-file
network *load_network(char *cfg, char *weights, int clear)
{
printf(" Try to load cfg: %s, weights: %s, clear = %d \n", cfg, weights, clear);
network *net = calloc(1, sizeof(network));
*net = parse_network_cfg(cfg);
if (weights && weights[0] != 0) {
load_weights(net, weights);
}
if (clear) (*net->seen) = 0;
return net;
}