From 336a19f14550ab1adbb0d9599284ac525fb6a5e0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 08 Sep 2016 05:44:41 +0000
Subject: [PATCH] slight changes to default demo settings
---
src/yolo.c | 11 ++++++++---
1 files changed, 8 insertions(+), 3 deletions(-)
diff --git a/src/yolo.c b/src/yolo.c
index 62455f2..65a7dd1 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -54,6 +54,11 @@
args.d = &buffer;
args.type = REGION_DATA;
+ args.angle = net.angle;
+ args.exposure = net.exposure;
+ args.saturation = net.saturation;
+ args.hue = net.hue;
+
pthread_t load_thread = load_data_in_thread(args);
clock_t time;
//while(i*imgs < N*120){
@@ -173,7 +178,7 @@
int nms = 1;
float iou_thresh = .5;
- int nthreads = 2;
+ int nthreads = 8;
image *val = calloc(nthreads, sizeof(image));
image *val_resized = calloc(nthreads, sizeof(image));
image *buf = calloc(nthreads, sizeof(image));
@@ -323,7 +328,7 @@
char buff[256];
char *input = buff;
int j;
- float nms=.5;
+ float nms=.4;
box *boxes = calloc(l.side*l.side*l.n, sizeof(box));
float **probs = calloc(l.side*l.side*l.n, sizeof(float *));
for(j = 0; j < l.side*l.side*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *));
@@ -370,7 +375,7 @@
voc_labels[i] = load_image_color(buff, 0, 0);
}
- float thresh = find_float_arg(argc, argv, "-thresh", .2);
+ float thresh = find_float_arg(argc, argv, "-thresh", .15);
int cam_index = find_int_arg(argc, argv, "-c", 0);
int frame_skip = find_int_arg(argc, argv, "-s", 0);
if(argc < 4){
--
Gitblit v1.10.0