From c40cdeb4021fc1a638969563972f13c9f5e90d74 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 09 Oct 2015 19:50:43 +0000
Subject: [PATCH] lots of comparator stuff
---
src/darknet.c | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index f87afc6..073156b 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -13,11 +13,14 @@
extern void run_imagenet(int argc, char **argv);
extern void run_yolo(int argc, char **argv);
+extern void run_swag(int argc, char **argv);
extern void run_coco(int argc, char **argv);
extern void run_writing(int argc, char **argv);
extern void run_captcha(int argc, char **argv);
extern void run_nightmare(int argc, char **argv);
extern void run_dice(int argc, char **argv);
+extern void run_compare(int argc, char **argv);
+extern void run_classifier(int argc, char **argv);
void change_rate(char *filename, float scale, float add)
{
@@ -86,7 +89,7 @@
if(weightfile){
load_weights_upto(&net, weightfile, max);
}
- net.seen = 0;
+ *net.seen = 0;
save_weights_upto(net, outfile, max);
}
@@ -177,8 +180,14 @@
average(argc, argv);
} else if (0 == strcmp(argv[1], "yolo")){
run_yolo(argc, argv);
+ } else if (0 == strcmp(argv[1], "swag")){
+ run_swag(argc, argv);
} else if (0 == strcmp(argv[1], "coco")){
run_coco(argc, argv);
+ } else if (0 == strcmp(argv[1], "classifier")){
+ run_classifier(argc, argv);
+ } else if (0 == strcmp(argv[1], "compare")){
+ run_compare(argc, argv);
} else if (0 == strcmp(argv[1], "dice")){
run_dice(argc, argv);
} else if (0 == strcmp(argv[1], "writing")){
--
Gitblit v1.10.0