| | |
| | | #include "opencv2/highgui/highgui_c.h" |
| | | #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 run_voxel(int argc, char **argv); |
| | | extern void run_yolo(int argc, char **argv); |
| | | extern void run_detector(int argc, char **argv); |
| | |
| | | run_super(argc, argv); |
| | | } 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", .24); |
| | | char *filename = (argc > 4) ? argv[4]: 0; |
| | | test_detector("cfg/coco.data", argv[2], argv[3], filename, thresh); |
| | | } else if (0 == strcmp(argv[1], "cifar")){ |
| | | run_cifar(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "go")){ |
| | |
| | | run_vid_rnn(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "coco")){ |
| | | run_coco(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "classify")){ |
| | | predict_classifier("cfg/imagenet1k.data", argv[2], argv[3], argv[4], 5); |
| | | } else if (0 == strcmp(argv[1], "classifier")){ |
| | | run_classifier(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "art")){ |