From 655f636a42d6e1d4518b712cfac6d973424de693 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 08 Mar 2015 18:25:28 +0000
Subject: [PATCH] detection layer fixed
---
src/parser.c | 9 +++++++--
1 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 53e1f56..0ee73a1 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -775,7 +775,7 @@
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 @@
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 @@
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");
--
Gitblit v1.10.0