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

Some bug fixes, random stuff

Browse Source
This commit is contained in:
Joseph Redmon
2015-09-01 11:21:01 -07:00
parent 9d42f49a24
commit 8bcdee8658
30 changed files with 629 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/parser.c
View File

@ -180,7 +180,8 @@ region_layer parse_region(list *options, size_params params)
int classes = option_find_int(options, "classes", 1);
int rescore = option_find_int(options, "rescore", 0);
int num = option_find_int(options, "num", 1);
region_layer layer = make_region_layer(params.batch, params.inputs, num, classes, coords, rescore);
int side = option_find_int(options, "side", 7);
region_layer layer = make_region_layer(params.batch, params.inputs, num, side, classes, coords, rescore);
return layer;
}
@ -342,6 +343,7 @@ network parse_network_cfg(char *filename)
n = n->next;
int count = 0;
free_section(s);
while(n){
fprintf(stderr, "%d: ", count);
s = (section *)n->val;
@ -521,6 +523,45 @@ list *read_cfg(char *filename)
return sections;
}
void save_weights_double(network net, char *filename)
{
fprintf(stderr, "Saving doubled weights to %s\n", filename);
FILE *fp = fopen(filename, "w");
if(!fp) file_error(filename);
fwrite(&net.learning_rate, sizeof(float), 1, fp);
fwrite(&net.momentum, sizeof(float), 1, fp);
fwrite(&net.decay, sizeof(float), 1, fp);
fwrite(&net.seen, sizeof(int), 1, fp);
int i,j,k;
for(i = 0; i < net.n; ++i){
layer l = net.layers[i];
if(l.type == CONVOLUTIONAL){
#ifdef GPU
if(gpu_index >= 0){
pull_convolutional_layer(l);
}
#endif
float zero = 0;
fwrite(l.biases, sizeof(float), l.n, fp);
fwrite(l.biases, sizeof(float), l.n, fp);
for (j = 0; j < l.n; ++j){
int index = j*l.c*l.size*l.size;
fwrite(l.filters+index, sizeof(float), l.c*l.size*l.size, fp);
for (k = 0; k < l.c*l.size*l.size; ++k) fwrite(&zero, sizeof(float), 1, fp);
}
for (j = 0; j < l.n; ++j){
int index = j*l.c*l.size*l.size;
for (k = 0; k < l.c*l.size*l.size; ++k) fwrite(&zero, sizeof(float), 1, fp);
fwrite(l.filters+index, sizeof(float), l.c*l.size*l.size, fp);
}
}
}
fclose(fp);
}
void save_weights_upto(network net, char *filename, int cutoff)
{
fprintf(stderr, "Saving weights to %s\n", filename);