|
|
|
@ -372,6 +372,7 @@ void run_coco(int argc, char **argv) |
|
|
|
|
char *out_filename = find_char_arg(argc, argv, "-out_filename", 0); |
|
|
|
|
char *prefix = find_char_arg(argc, argv, "-prefix", 0); |
|
|
|
|
float thresh = find_float_arg(argc, argv, "-thresh", .2); |
|
|
|
|
float hier_thresh = find_float_arg(argc, argv, "-hier", .5); |
|
|
|
|
int cam_index = find_int_arg(argc, argv, "-c", 0); |
|
|
|
|
int frame_skip = find_int_arg(argc, argv, "-s", 0); |
|
|
|
|
|
|
|
|
@ -387,6 +388,6 @@ void run_coco(int argc, char **argv) |
|
|
|
|
else if(0==strcmp(argv[2], "train")) train_coco(cfg, weights); |
|
|
|
|
else if(0==strcmp(argv[2], "valid")) validate_coco(cfg, weights); |
|
|
|
|
else if(0==strcmp(argv[2], "recall")) validate_coco_recall(cfg, weights); |
|
|
|
|
else if(0==strcmp(argv[2], "demo")) demo(cfg, weights, thresh, cam_index, filename, coco_classes, 80, frame_skip,
|
|
|
|
|
else if(0==strcmp(argv[2], "demo")) demo(cfg, weights, thresh, hier_thresh, cam_index, filename, coco_classes, 80, frame_skip, |
|
|
|
|
prefix, out_filename, http_stream_port, dont_show); |
|
|
|
|
} |
|
|
|
|