From cf32e7e9b843560eb7ec3ed16e5b19f0f7156724 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@burninator.cs.washington.edu>
Date: Sat, 25 Jun 2016 23:12:00 +0000
Subject: [PATCH] colors

---
 src/dice.c |    8 +++++---
 1 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/dice.c b/src/dice.c
index 7948741..6f148b0 100644
--- a/src/dice.c
+++ b/src/dice.c
@@ -61,7 +61,7 @@
     free_list(plist);
 
     data val = load_data(paths, m, 0, labels, 6, net.w, net.h);
-    float *acc = network_accuracies(net, val);
+    float *acc = network_accuracies(net, val, 2);
     printf("Validation Accuracy: %f, %d images\n", acc[0], m);
     free_data(val);
 }
@@ -76,7 +76,8 @@
     srand(2222222);
     int i = 0;
     char **names = dice_labels;
-    char input[256];
+    char buff[256];
+    char *input = buff;
     int indexes[6];
     while(1){
         if(filename){
@@ -84,7 +85,8 @@
         }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, net.w, net.h);

--
Gitblit v1.10.0