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/classifier.c | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index c0006e6..8a3ae5a 100644
--- a/src/classifier.c
+++ b/src/classifier.c
@@ -131,7 +131,7 @@
char *label_list = option_find_str(options, "labels", "data/labels.list");
char *valid_list = option_find_str(options, "valid", "data/train.list");
int classes = option_find_int(options, "classes", 2);
- int topk = option_find_int(options, "topk", 1);
+ int topk = option_find_int(options, "top", 1);
char **labels = get_labels(label_list);
list *plist = get_paths(valid_list);
@@ -194,11 +194,12 @@
list *options = read_data_cfg(datacfg);
- char *label_list = option_find_str(options, "labels", "data/labels.list");
+ char *name_list = option_find_str(options, "names", 0);
+ if(!name_list) name_list = option_find_str(options, "labels", "data/labels.list");
int top = option_find_int(options, "top", 1);
int i = 0;
- char **names = get_labels(label_list);
+ char **names = get_labels(name_list);
clock_t time;
int indexes[10];
char buff[256];
--
Gitblit v1.10.0