| | |
| | | |
| | | extern void run_imagenet(int argc, char **argv); |
| | | extern void run_yolo(int argc, char **argv); |
| | | extern void run_swag(int argc, char **argv); |
| | | extern void run_coco(int argc, char **argv); |
| | | extern void run_writing(int argc, char **argv); |
| | | extern void run_captcha(int argc, char **argv); |
| | | extern void run_nightmare(int argc, char **argv); |
| | | extern void run_dice(int argc, char **argv); |
| | | extern void run_compare(int argc, char **argv); |
| | | extern void run_classifier(int argc, char **argv); |
| | | |
| | | void change_rate(char *filename, float scale, float add) |
| | | { |
| | |
| | | if(weightfile){ |
| | | load_weights_upto(&net, weightfile, max); |
| | | } |
| | | net.seen = 0; |
| | | *net.seen = 0; |
| | | save_weights_upto(net, outfile, max); |
| | | } |
| | | |
| | |
| | | average(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "yolo")){ |
| | | run_yolo(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "swag")){ |
| | | run_swag(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "coco")){ |
| | | run_coco(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "classifier")){ |
| | | run_classifier(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "compare")){ |
| | | run_compare(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "dice")){ |
| | | run_dice(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "writing")){ |