From 5a49c1d962895b4b2835ae47201d6b07e669153b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 10 Apr 2015 06:00:33 +0000
Subject: [PATCH] rotation
---
src/data.c | 8 ++++++--
1 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/data.c b/src/data.c
index c454d84..34108f6 100644
--- a/src/data.c
+++ b/src/data.c
@@ -57,6 +57,12 @@
for(i = 0; i < n; ++i){
image im = load_image_color(paths[i], w, h);
+ translate_image(im, -128);
+ scale_image(im, 1./128);
+ float rad = rand_uniform() - .5;
+ image rot = rotate_image(im, rad);
+ free_image(im);
+ im = rot;
X.vals[i] = im.data;
X.cols = im.h*im.w*im.c;
}
@@ -373,8 +379,6 @@
{
struct load_args a = *(struct load_args*)ptr;
*a.d = load_data(a.paths, a.n, a.m, a.labels, a.k, a.w, a.h);
- translate_data_rows(*a.d, -128);
- scale_data_rows(*a.d, 1./128);
free(ptr);
return 0;
}
--
Gitblit v1.10.0