| 383 | } |
| 384 | |
| 385 | int 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); |
nothing calls this directly
no test coverage detected