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

detection layer fixed

Browse Source
This commit is contained in:
Joseph Redmon
2015-03-08 11:25:28 -07:00
parent 2313a8eb54
commit 655f636a42
9 changed files with 141 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/parser.c
View File

@ -775,7 +775,7 @@ void save_weights(network net, char *filename)
fclose(fp);
}
void load_weights(network *net, char *filename)
void load_weights_upto(network *net, char *filename, int cutoff)
{
fprintf(stderr, "Loading weights from %s\n", filename);
FILE *fp = fopen(filename, "r");
@ -788,7 +788,7 @@ void load_weights(network *net, char *filename)
set_learning_network(net, net->learning_rate, net->momentum, net->decay);
int i;
for(i = 0; i < net->n; ++i){
for(i = 0; i < net->n && i < cutoff; ++i){
if(net->types[i] == CONVOLUTIONAL){
convolutional_layer layer = *(convolutional_layer *) net->layers[i];
int num = layer.n*layer.c*layer.size*layer.size;
@ -825,6 +825,11 @@ void load_weights(network *net, char *filename)
fclose(fp);
}
void load_weights(network *net, char *filename)
{
load_weights_upto(net, filename, net->n);
}
void save_network(network net, char *filename)
{
FILE *fp = fopen(filename, "w");