From 79fffcce3ce495bd415dc1284224c915d7194d4c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Dec 2014 21:15:26 +0000
Subject: [PATCH] Better imagenet distributed training

---
 src/data.h |    3 +++
 1 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/src/data.h b/src/data.h
index b9707c9..38a5e15 100644
--- a/src/data.h
+++ b/src/data.h
@@ -1,5 +1,6 @@
 #ifndef DATA_H
 #define DATA_H
+#include <pthread.h>
 
 #include "matrix.h"
 #include "list.h"
@@ -13,8 +14,10 @@
 
 void free_data(data d);
 data load_data(char **paths, int n, char **labels, int k, int h, int w);
+pthread_t load_data_random_thread(int n, char **paths, int m, char **labels, int k, int h, int w, data *d);
 data load_data_random(int n, char **paths, int m, char **labels, int k, int h, int w);
 data load_data_detection_random(int n, char **paths, int m, int h, int w, int nh, int nw, float scale);
+data load_data_detection_jitter_random(int n, char **paths, int m, int h, int w, int nh, int nw, float scale);
 data load_data_image_pathfile(char *filename, char **labels, int k, int h, int w);
 data load_cifar10_data(char *filename);
 data load_all_cifar10();

--
Gitblit v1.10.0