From aebe937710ced03d03f73ab23f410f29685655c1 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Aug 2016 18:54:24 +0000
Subject: [PATCH] what do you even write here?
---
src/classifier.c | 21 ++++++++++++---------
1 files changed, 12 insertions(+), 9 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index 608e3ab..ee6d212 100644
--- a/src/classifier.c
+++ b/src/classifier.c
@@ -41,7 +41,7 @@
void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int clear)
{
- int nthreads = 2;
+ int nthreads = 8;
int i;
data_seed = time(0);
@@ -82,6 +82,9 @@
args.min = net.min_crop;
args.max = net.max_crop;
+ args.angle = net.angle;
+ args.exposure = net.exposure;
+ args.saturation = net.saturation;
args.size = net.w;
args.paths = paths;
@@ -113,14 +116,14 @@
printf("Loaded: %lf seconds\n", sec(clock()-time));
time=clock();
- /*
- int u;
- for(u = 0; u < net.batch; ++u){
- image im = float_to_image(net.w, net.h, 3, train.X.vals[u]);
- show_image(im, "loaded");
- cvWaitKey(0);
- }
- */
+ if(0){
+ int u;
+ for(u = 0; u < imgs; ++u){
+ image im = float_to_image(net.w, net.h, 3, train.X.vals[u]);
+ show_image(im, "loaded");
+ cvWaitKey(0);
+ }
+ }
float loss = train_network(net, train);
if(avg_loss == -1) avg_loss = loss;
--
Gitblit v1.10.0