From f996bd59a61338d8d51e2b19482d684f6dd04d0f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 23 Sep 2015 21:16:48 +0000
Subject: [PATCH] more writing fixes
---
src/darknet.c | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index f87afc6..9632f91 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -13,11 +13,13 @@
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);
void change_rate(char *filename, float scale, float add)
{
@@ -86,7 +88,7 @@
if(weightfile){
load_weights_upto(&net, weightfile, max);
}
- net.seen = 0;
+ *net.seen = 0;
save_weights_upto(net, outfile, max);
}
@@ -177,8 +179,12 @@
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], "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