From b8e6e80c6d411d05a9e09f1e3676eb9a7f3ea0e8 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 03 Aug 2018 11:35:03 +0000
Subject: [PATCH] Added spatial Yolo v3 yolov3-spp.cfg
---
src/option_list.c | 21 ++++++++++++++++++++-
1 files changed, 20 insertions(+), 1 deletions(-)
diff --git a/src/option_list.c b/src/option_list.c
index f935af3..dfab81b 100644
--- a/src/option_list.c
+++ b/src/option_list.c
@@ -12,7 +12,7 @@
int nu = 0;
list *options = make_list();
while((line=fgetl(file)) != 0){
- ++ nu;
+ ++nu;
strip(line);
switch(line[0]){
case '\0':
@@ -32,6 +32,25 @@
return options;
}
+metadata get_metadata(char *file)
+{
+ metadata m = { 0 };
+ list *options = read_data_cfg(file);
+
+ char *name_list = option_find_str(options, "names", 0);
+ if (!name_list) name_list = option_find_str(options, "labels", 0);
+ if (!name_list) {
+ fprintf(stderr, "No names or labels found\n");
+ }
+ else {
+ m.names = get_labels(name_list);
+ }
+ m.classes = option_find_int(options, "classes", 2);
+ free_list(options);
+ printf("Loaded - names_list: %s, classes = %d \n", name_list, m.classes);
+ return m;
+}
+
int read_option(char *s, list *options)
{
size_t i;
--
Gitblit v1.10.0