From 8fd18add6e060a433629fae3fa2a7ef75df4644e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 04 Nov 2015 03:23:42 +0000
Subject: [PATCH] CVPR Experiments

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

diff --git a/src/imagenet.c b/src/imagenet.c
index c826a0f..fa16251 100644
--- a/src/imagenet.c
+++ b/src/imagenet.c
@@ -92,6 +92,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 +134,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);
@@ -152,15 +153,17 @@
     int i = 0;
     char **names = get_labels("data/shortnames.txt");
     clock_t time;
-    char input[256];
     int indexes[10];
+    char buff[256];
+    char *input = buff;
     while(1){
         if(filename){
             strncpy(input, filename, 256);
         }else{
             printf("Enter Image Path: ");
             fflush(stdout);
-            fgets(input, 256, stdin);
+            input = fgets(input, 256, stdin);
+            if(!input) return;
             strtok(input, "\n");
         }
         image im = load_image_color(input, 256, 256);

--
Gitblit v1.10.0