From 2db9fbef2bd7d35a547d0018a9850f6b249c524f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 13 Nov 2013 18:50:38 +0000
Subject: [PATCH] Parsing, image loading, lots of stuff
---
src/image.c | 29 +++++++++++++++++++++++++++--
1 files changed, 27 insertions(+), 2 deletions(-)
diff --git a/src/image.c b/src/image.c
index a1aa8a7..a509d32 100644
--- a/src/image.c
+++ b/src/image.c
@@ -34,6 +34,18 @@
p.data[i+j*p.h*p.w] = (p.data[i+j*p.h*p.w] - min[j])/(max[j]-min[j]);
}
}
+ free(min);
+ free(max);
+}
+
+double avg_image_layer(image m, int l)
+{
+ int i;
+ double sum = 0;
+ for(i = 0; i < m.h*m.w; ++i){
+ sum += m.data[l*m.h*m.w + i];
+ }
+ return sum/(m.h*m.w);
}
void threshold_image(image p, double t)
@@ -95,16 +107,29 @@
}
}
-image make_image(int h, int w, int c)
+image make_empty_image(int h, int w, int c)
{
image out;
out.h = h;
out.w = w;
out.c = c;
+ return out;
+}
+
+image make_image(int h, int w, int c)
+{
+ image out = make_empty_image(h,w,c);
out.data = calloc(h*w*c, sizeof(double));
return out;
}
+image double_to_image(int h, int w, int c, double *data)
+{
+ image out = make_empty_image(h,w,c);
+ out.data = data;
+ return out;
+}
+
void zero_image(image m)
{
memset(m.data, 0, m.h*m.w*m.c*sizeof(double));
@@ -132,7 +157,7 @@
image out = make_image(h,w,c);
int i;
for(i = 0; i < h*w*c; ++i){
- out.data[i] = .5-(double)rand()/RAND_MAX;
+ out.data[i] = (.5-(double)rand()/RAND_MAX);
}
return out;
}
--
Gitblit v1.10.0