| | |
| | | #endif |
| | | |
| | | extern void predict_classifier(char *datacfg, char *cfgfile, char *weightfile, char *filename, int top); |
| | | extern void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh); |
| | | extern void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, int ext_output); |
| | | extern void run_voxel(int argc, char **argv); |
| | | extern void run_yolo(int argc, char **argv); |
| | | extern void run_detector(int argc, char **argv); |
| | |
| | | |
| | | int main(int argc, char **argv) |
| | | { |
| | | #ifdef _DEBUG |
| | | _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF); |
| | | #endif |
| | | |
| | | int i; |
| | | for (i = 0; i < argc; ++i) { |
| | | if (!argv[i]) continue; |
| | | strip_args(argv[i]); |
| | | } |
| | | |
| | | //test_resize("data/bad.jpg"); |
| | | //test_box(); |
| | | //test_convolutional_layer(); |
| | |
| | | } else if (0 == strcmp(argv[1], "detector")){ |
| | | run_detector(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "detect")){ |
| | | float thresh = find_float_arg(argc, argv, "-thresh", .25); |
| | | float thresh = find_float_arg(argc, argv, "-thresh", .24); |
| | | int ext_output = find_arg(argc, argv, "-ext_output"); |
| | | char *filename = (argc > 4) ? argv[4]: 0; |
| | | test_detector("cfg/coco.data", argv[2], argv[3], filename, thresh); |
| | | test_detector("cfg/coco.data", argv[2], argv[3], filename, thresh, ext_output); |
| | | } else if (0 == strcmp(argv[1], "cifar")){ |
| | | run_cifar(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "go")){ |