From d0b9326a352ed2fbc3ae66fdef40b4533a2f211d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 11 Aug 2015 06:22:27 +0000
Subject: [PATCH] Hacks to get nightmare to not break gridsizing
---
src/darknet.c | 89 ++++++++++++++++++++++++--------------------
1 files changed, 49 insertions(+), 40 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index 411efdf..d7fb1f5 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -11,42 +11,10 @@
extern void run_imagenet(int argc, char **argv);
extern void run_detection(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);
-
-void del_arg(int argc, char **argv, int index)
-{
- int i;
- for(i = index; i < argc-1; ++i) argv[i] = argv[i+1];
- argv[i] = 0;
-}
-
-int find_arg(int argc, char* argv[], char *arg)
-{
- int i;
- for(i = 0; i < argc; ++i) {
- if(!argv[i]) continue;
- if(0==strcmp(argv[i], arg)) {
- del_arg(argc, argv, i);
- return 1;
- }
- }
- return 0;
-}
-
-int find_int_arg(int argc, char **argv, char *arg, int def)
-{
- int i;
- for(i = 0; i < argc-1; ++i){
- if(!argv[i]) continue;
- if(0==strcmp(argv[i], arg)){
- def = atoi(argv[i+1]);
- del_arg(argc, argv, i);
- del_arg(argc, argv, i);
- break;
- }
- }
- return def;
-}
+extern void run_nightmare(int argc, char **argv);
void change_rate(char *filename, float scale, float add)
{
@@ -69,16 +37,44 @@
load_weights_upto(&net, weightfile, max);
}
net.seen = 0;
- save_weights(net, outfile);
+ save_weights_upto(net, outfile, max);
}
-void convert(char *cfgfile, char *outfile, char *weightfile)
+#include "convolutional_layer.h"
+void rescale_net(char *cfgfile, char *weightfile, char *outfile)
{
+ gpu_index = -1;
network net = parse_network_cfg(cfgfile);
if(weightfile){
load_weights(&net, weightfile);
}
- save_network(net, outfile);
+ int i;
+ for(i = 0; i < net.n; ++i){
+ layer l = net.layers[i];
+ if(l.type == CONVOLUTIONAL){
+ rescale_filters(l, 2, -.5);
+ break;
+ }
+ }
+ save_weights(net, outfile);
+}
+
+void rgbgr_net(char *cfgfile, char *weightfile, char *outfile)
+{
+ gpu_index = -1;
+ network net = parse_network_cfg(cfgfile);
+ if(weightfile){
+ load_weights(&net, weightfile);
+ }
+ int i;
+ for(i = 0; i < net.n; ++i){
+ layer l = net.layers[i];
+ if(l.type == CONVOLUTIONAL){
+ rgbgr_filters(l);
+ break;
+ }
+ }
+ save_weights(net, outfile);
}
void visualize(char *cfgfile, char *weightfile)
@@ -88,11 +84,14 @@
load_weights(&net, weightfile);
}
visualize_network(net);
+ #ifdef OPENCV
cvWaitKey(0);
+ #endif
}
int main(int argc, char **argv)
{
+ //test_resize("data/bad.jpg");
//test_box();
//test_convolutional_layer();
if(argc < 2){
@@ -114,18 +113,28 @@
run_imagenet(argc, argv);
} else if (0 == strcmp(argv[1], "detection")){
run_detection(argc, argv);
+ } else if (0 == strcmp(argv[1], "coco")){
+ run_coco(argc, argv);
+ } else if (0 == strcmp(argv[1], "writing")){
+ run_writing(argc, argv);
} else if (0 == strcmp(argv[1], "test")){
test_resize(argv[2]);
} else if (0 == strcmp(argv[1], "captcha")){
run_captcha(argc, argv);
+ } else if (0 == strcmp(argv[1], "nightmare")){
+ run_nightmare(argc, argv);
} else if (0 == strcmp(argv[1], "change")){
change_rate(argv[2], atof(argv[3]), (argc > 4) ? atof(argv[4]) : 0);
- } else if (0 == strcmp(argv[1], "convert")){
- convert(argv[2], argv[3], (argc > 4) ? argv[4] : 0);
+ } else if (0 == strcmp(argv[1], "rgbgr")){
+ rgbgr_net(argv[2], argv[3], argv[4]);
+ } else if (0 == strcmp(argv[1], "rescale")){
+ rescale_net(argv[2], argv[3], argv[4]);
} else if (0 == strcmp(argv[1], "partial")){
partial(argv[2], argv[3], argv[4], atoi(argv[5]));
} else if (0 == strcmp(argv[1], "visualize")){
visualize(argv[2], (argc > 3) ? argv[3] : 0);
+ } else if (0 == strcmp(argv[1], "imtest")){
+ test_resize(argv[2]);
} else {
fprintf(stderr, "Not an option: %s\n", argv[1]);
}
--
Gitblit v1.10.0