From 989ab8c38a02fa7ea9c25108151736c62e81c972 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 24 Apr 2015 17:27:50 +0000
Subject: [PATCH] IOU loss function
---
src/darknet.c | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index 0cd6153..411efdf 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -68,7 +68,7 @@
if(weightfile){
load_weights_upto(&net, weightfile, max);
}
- //net.seen = 0;
+ net.seen = 0;
save_weights(net, outfile);
}
@@ -93,7 +93,7 @@
int main(int argc, char **argv)
{
- //test_resize(argv[1]);
+ //test_box();
//test_convolutional_layer();
if(argc < 2){
fprintf(stderr, "usage: %s <function>\n", argv[0]);
@@ -114,6 +114,8 @@
run_imagenet(argc, argv);
} else if (0 == strcmp(argv[1], "detection")){
run_detection(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], "change")){
--
Gitblit v1.10.0