MCPcopy Create free account
hub / github.com/OpenPTrack/open_ptrack_v2 / main

Function main

yolo_detector/darknet_opt/examples/darknet.c:385–487  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

383}
384
385int main(int argc, char **argv)
386{
387 //test_resize("data/bad.jpg");
388 //test_box();
389 //test_convolutional_layer();
390 if(argc < 2){
391 fprintf(stderr, "usage: %s <function>\n", argv[0]);
392 return 0;
393 }
394 gpu_index = find_int_arg(argc, argv, "-i", 0);
395 if(find_arg(argc, argv, "-nogpu")) {
396 gpu_index = -1;
397 }
398
399#ifndef GPU
400 gpu_index = -1;
401#else
402 if(gpu_index >= 0){
403 cuda_set_device(gpu_index);
404 }
405#endif
406
407 if (0 == strcmp(argv[1], "average")){
408 average(argc, argv);
409 } else if (0 == strcmp(argv[1], "yolo")){
410 run_yolo(argc, argv);
411 } else if (0 == strcmp(argv[1], "super")){
412 run_super(argc, argv);
413 } else if (0 == strcmp(argv[1], "lsd")){
414 run_lsd(argc, argv);
415 } else if (0 == strcmp(argv[1], "detector")){
416 run_detector(argc, argv);
417 } else if (0 == strcmp(argv[1], "detect")){
418 float thresh = find_float_arg(argc, argv, "-thresh", .24);
419 char *filename = (argc > 4) ? argv[4]: 0;
420 char *outfile = find_char_arg(argc, argv, "-out", 0);
421 int fullscreen = find_arg(argc, argv, "-fullscreen");
422 test_detector("cfg/coco.data", argv[2], argv[3], filename, thresh, .5, outfile, fullscreen);
423 } else if (0 == strcmp(argv[1], "cifar")){
424 run_cifar(argc, argv);
425 } else if (0 == strcmp(argv[1], "go")){
426 run_go(argc, argv);
427 } else if (0 == strcmp(argv[1], "rnn")){
428 run_char_rnn(argc, argv);
429 } else if (0 == strcmp(argv[1], "coco")){
430 run_coco(argc, argv);
431 } else if (0 == strcmp(argv[1], "classify")){
432 predict_classifier("cfg/imagenet1k.data", argv[2], argv[3], argv[4], 5);
433 } else if (0 == strcmp(argv[1], "classifier")){
434 run_classifier(argc, argv);
435 } else if (0 == strcmp(argv[1], "attention")){
436 run_attention(argc, argv);
437 } else if (0 == strcmp(argv[1], "regressor")){
438 run_regressor(argc, argv);
439 } else if (0 == strcmp(argv[1], "segmenter")){
440 run_segmenter(argc, argv);
441 } else if (0 == strcmp(argv[1], "art")){
442 run_art(argc, argv);

Callers

nothing calls this directly

Calls 15

find_int_argFunction · 0.85
find_argFunction · 0.85
cuda_set_deviceFunction · 0.85
averageFunction · 0.85
run_yoloFunction · 0.85
run_superFunction · 0.85
run_lsdFunction · 0.85
run_detectorFunction · 0.85
find_float_argFunction · 0.85
find_char_argFunction · 0.85
test_detectorFunction · 0.85
run_cifarFunction · 0.85

Tested by

no test coverage detected