mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

slight changes to default demo settings

Browse Source
This commit is contained in:
Joseph Redmon 2016-09-07 22:44:41 -07:00
parent 6b38dcdce0
commit 336a19f145
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/demo.c
View File

@ -48,7 +48,7 @@ void *fetch_in_thread(void *ptr)
void *detect_in_thread(void *ptr) void *detect_in_thread(void *ptr)
{ {
float nms = .1; float nms = .4;
detection_layer l = net.layers[net.n-1]; detection_layer l = net.layers[net.n-1];
float *X = det_s.data; float *X = det_s.data;

6
src/yolo.c
View File

@ -178,7 +178,7 @@ void validate_yolo(char *cfgfile, char *weightfile)
int nms = 1; int nms = 1;
float iou_thresh = .5; float iou_thresh = .5;
int nthreads = 2; int nthreads = 8;
image *val = calloc(nthreads, sizeof(image)); image *val = calloc(nthreads, sizeof(image));
image *val_resized = calloc(nthreads, sizeof(image)); image *val_resized = calloc(nthreads, sizeof(image));
image *buf = calloc(nthreads, sizeof(image)); image *buf = calloc(nthreads, sizeof(image));
@ -328,7 +328,7 @@ void test_yolo(char *cfgfile, char *weightfile, char *filename, float thresh)
char buff[256]; char buff[256];
char *input = buff; char *input = buff;
int j; int j;
float nms=.5; float nms=.4;
box *boxes = calloc(l.side*l.side*l.n, sizeof(box)); box *boxes = calloc(l.side*l.side*l.n, sizeof(box));
float **probs = calloc(l.side*l.side*l.n, sizeof(float *)); 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 *)); for(j = 0; j < l.side*l.side*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *));
@ -375,7 +375,7 @@ void run_yolo(int argc, char **argv)
voc_labels[i] = load_image_color(buff, 0, 0); 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 cam_index = find_int_arg(argc, argv, "-c", 0);
int frame_skip = find_int_arg(argc, argv, "-s", 0); int frame_skip = find_int_arg(argc, argv, "-s", 0);
if(argc < 4){ if(argc < 4){