From b13ad6d5fd23f68f506c14ede4282126d893702b Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 05 Nov 2014 22:49:58 +0000
Subject: [PATCH] Can validate on imagenet now
---
src/data.c | 14 +++++++++++++-
1 files changed, 13 insertions(+), 1 deletions(-)
diff --git a/src/data.c b/src/data.c
index 734fffa..a5da9d3 100644
--- a/src/data.c
+++ b/src/data.c
@@ -83,6 +83,7 @@
data load_data_image_pathfile_part(char *filename, int part, int total, char **labels, int k, int h, int w)
{
+ clock_t time = clock();
list *plist = get_paths(filename);
char **paths = (char **)list_to_array(plist);
int start = part*plist->size/total;
@@ -172,7 +173,7 @@
return d;
}
-void get_batch(data d, int n, float *X, float *y)
+void get_random_batch(data d, int n, float *X, float *y)
{
int j;
for(j = 0; j < n; ++j){
@@ -182,6 +183,17 @@
}
}
+void get_next_batch(data d, int n, int offset, float *X, float *y)
+{
+ int j;
+ for(j = 0; j < n; ++j){
+ int index = offset + j;
+ memcpy(X+j*d.X.cols, d.X.vals[index], d.X.cols*sizeof(float));
+ memcpy(y+j*d.y.cols, d.y.vals[index], d.y.cols*sizeof(float));
+ }
+}
+
+
data load_all_cifar10()
{
data d;
--
Gitblit v1.10.0