From 37d7c1e79f65a75caf87e29a562d30c51cd654e5 Mon Sep 17 00:00:00 2001
From: Joe Redmon <pjreddie@gmail.com>
Date: Thu, 26 Nov 2015 21:52:56 +0000
Subject: [PATCH] fixed label linking
---
src/imagenet.c | 15 ++++++---------
1 files changed, 6 insertions(+), 9 deletions(-)
diff --git a/src/imagenet.c b/src/imagenet.c
index c826a0f..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);
@@ -152,15 +147,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