| | |
| | | #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, 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); |
| | |
| | | network net = parse_network_cfg(cfgfile); |
| | | int oldn = net.layers[net.n - 2].n; |
| | | int c = net.layers[net.n - 2].c; |
| | | net.layers[net.n - 2].n = 7879; |
| | | net.layers[net.n - 2].n = 9372; |
| | | net.layers[net.n - 2].biases += 5; |
| | | net.layers[net.n - 2].weights += 5*c; |
| | | if(weightfile){ |
| | |
| | | |
| | | 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(); |
| | |
| | | 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); |
| | | 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, ext_output); |
| | | } 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")){ |