From c7b10ceadb1a78e7480d281444a31ae2a7dc1b05 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 06 May 2016 23:25:16 +0000
Subject: [PATCH] so much need to commit
---
src/classifier.c | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index efed514..7060c5e 100644
--- a/src/classifier.c
+++ b/src/classifier.c
@@ -3,6 +3,7 @@
#include "parser.h"
#include "option_list.h"
#include "blas.h"
+#include "classifier.h"
#include <sys/time.h>
#ifdef OPENCV
@@ -49,7 +50,7 @@
load_weights(&net, weightfile);
}
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
- int imgs = 1024;
+ int imgs = net.batch;
list *options = read_data_cfg(datacfg);
@@ -72,7 +73,7 @@
args.w = net.w;
args.h = net.h;
- args.min = net.w;
+ args.min = net.min_crop;
args.max = net.max_crop;
args.size = net.w;
@@ -410,7 +411,7 @@
char **labels = get_labels(label_list);
list *plist = get_paths(valid_list);
- int scales[] = {160, 192, 224, 288, 320, 352, 384};
+ int scales[] = {192, 224, 288, 320, 352};
int nscales = sizeof(scales)/sizeof(scales[0]);
char **paths = (char **)list_to_array(plist);
--
Gitblit v1.10.0