From 38bd6ae6ba24fc8c14fd61d1238ae94a983434b3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 20 Jul 2015 23:16:26 +0000
Subject: [PATCH] Better partial function
---
src/parser.c | 14 ++++++++++----
1 files changed, 10 insertions(+), 4 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 893111e..5591dc3 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -167,7 +167,7 @@
int rescore = option_find_int(options, "rescore", 0);
int joint = option_find_int(options, "joint", 0);
int objectness = option_find_int(options, "objectness", 0);
- int background = option_find_int(options, "background", 0);
+ int background = 0;
detection_layer layer = make_detection_layer(params.batch, params.inputs, classes, coords, joint, rescore, background, objectness);
return layer;
}
@@ -196,7 +196,10 @@
batch=params.batch;
if(!(h && w && c)) error("Layer before crop layer must output image.");
+ int noadjust = option_find_int_quiet(options, "noadjust",0);
+
crop_layer l = make_crop_layer(batch,h,w,c,crop_height,crop_width,flip, angle, saturation, exposure);
+ l.noadjust = noadjust;
return l;
}
@@ -292,7 +295,6 @@
net->learning_rate = option_find_float(options, "learning_rate", .001);
net->momentum = option_find_float(options, "momentum", .9);
net->decay = option_find_float(options, "decay", .0001);
- net->seen = option_find_int(options, "seen",0);
int subdivs = option_find_int(options, "subdivisions",1);
net->batch /= subdivs;
net->subdivisions = subdivs;
@@ -498,7 +500,7 @@
return sections;
}
-void save_weights(network net, char *filename)
+void save_weights_upto(network net, char *filename, int cutoff)
{
fprintf(stderr, "Saving weights to %s\n", filename);
FILE *fp = fopen(filename, "w");
@@ -510,7 +512,7 @@
fwrite(&net.seen, sizeof(int), 1, fp);
int i;
- for(i = 0; i < net.n; ++i){
+ for(i = 0; i < net.n && i < cutoff; ++i){
layer l = net.layers[i];
if(l.type == CONVOLUTIONAL){
#ifdef GPU
@@ -544,6 +546,10 @@
}
fclose(fp);
}
+void save_weights(network net, char *filename)
+{
+ save_weights_upto(net, filename, net.n);
+}
void load_weights_upto(network *net, char *filename, int cutoff)
{
--
Gitblit v1.10.0