| | |
| | | void rgbgr_filters(convolutional_layer l); |
| | | void rgbgr_net(char *cfgfile, char *weightfile, char *outfile) |
| | | { |
| | | gpu_index = -1; |
| | | network net = parse_network_cfg(cfgfile); |
| | | if(weightfile){ |
| | | load_weights(&net, weightfile); |
| | |
| | | |
| | | int main(int argc, char **argv) |
| | | { |
| | | //test_resize("data/cat.png"); |
| | | //test_resize("data/bad.jpg"); |
| | | //test_box(); |
| | | //test_convolutional_layer(); |
| | | if(argc < 2){ |
| | |
| | | partial(argv[2], argv[3], argv[4], atoi(argv[5])); |
| | | } else if (0 == strcmp(argv[1], "visualize")){ |
| | | visualize(argv[2], (argc > 3) ? argv[3] : 0); |
| | | } else if (0 == strcmp(argv[1], "imtest")){ |
| | | test_resize(argv[2]); |
| | | } else { |
| | | fprintf(stderr, "Not an option: %s\n", argv[1]); |
| | | } |