From 86e6cb6278041492ce2281f7ed8d8dcea37e50c7 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@users.noreply.github.com>
Date: Wed, 05 Aug 2015 23:56:24 +0000
Subject: [PATCH] Update README.md
---
src/imagenet.c | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/imagenet.c b/src/imagenet.c
index 0e272fc..96e74d0 100644
--- a/src/imagenet.c
+++ b/src/imagenet.c
@@ -25,7 +25,7 @@
pthread_t load_thread;
data train;
data buffer;
- load_thread = load_data_thread(paths, imgs, plist->size, labels, 1000, 256, 256, &buffer);
+ load_thread = load_data_thread(paths, imgs, plist->size, labels, 1000, net.w, net.h, &buffer);
while(1){
++i;
time=clock();
@@ -38,7 +38,7 @@
cvWaitKey(0);
*/
- load_thread = load_data_thread(paths, imgs, plist->size, labels, 1000, 256, 256, &buffer);
+ load_thread = load_data_thread(paths, imgs, plist->size, labels, 1000, net.w, net.h, &buffer);
printf("Loaded: %lf seconds\n", sec(clock()-time));
time=clock();
float loss = train_network(net, train);
--
Gitblit v1.10.0