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 | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index 2e974a5..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;
--
Gitblit v1.10.0