From 8c5364f58569eaeb5582a4915b36b24fc5570c76 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 09 Nov 2015 19:31:39 +0000
Subject: [PATCH] New YOLO
---
src/imagenet.c | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/src/imagenet.c b/src/imagenet.c
index 567a8c4..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);
--
Gitblit v1.10.0