From db0397cfaaf488364e3d2e1669dfefae2ee6ea73 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 14 Dec 2015 19:57:10 +0000
Subject: [PATCH] shortcut layers, msr networks

---
 src/imagenet.c |    9 ++-------
 1 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/imagenet.c b/src/imagenet.c
index 567a8c4..dece952 100644
--- a/src/imagenet.c
+++ b/src/imagenet.c
@@ -48,12 +48,6 @@
         pthread_join(load_thread, 0);
         train = buffer;
 
-        /*
-        image im = float_to_image(256, 256, 3, train.X.vals[114]);
-        show_image(im, "training");
-        cvWaitKey(0);
-        */
-
         load_thread = load_data_in_thread(args);
         printf("Loaded: %lf seconds\n", sec(clock()-time));
         time=clock();
@@ -92,6 +86,7 @@
     srand(time(0));
 
     char **labels = get_labels("data/inet.labels.list");
+    //list *plist = get_paths("data/inet.suppress.list");
     list *plist = get_paths("data/inet.val.list");
 
     char **paths = (char **)list_to_array(plist);
@@ -133,7 +128,7 @@
         printf("Loaded: %d images in %lf seconds\n", val.X.rows, sec(clock()-time));
 
         time=clock();
-        float *acc = network_accuracies(net, val);
+        float *acc = network_accuracies(net, val, 5);
         avg_acc += acc[0];
         avg_top5 += acc[1];
         printf("%d: top1: %f, top5: %f, %lf seconds, %d images\n", i, avg_acc/i, avg_top5/i, sec(clock()-time), val.X.rows);

--
Gitblit v1.10.0