| | |
| | | #include "utils.h" |
| | | #include "cuda.h" |
| | | |
| | | #define _GNU_SOURCE |
| | | #include <fenv.h> |
| | | #ifdef OPENCV |
| | | #include "opencv2/highgui/highgui_c.h" |
| | | #endif |
| | | |
| | | extern void run_imagenet(int argc, char **argv); |
| | | extern void run_detection(int argc, char **argv); |
| | | extern void run_yolo(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); |
| | |
| | | run_imagenet(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "detection")){ |
| | | run_detection(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "yolo")){ |
| | | run_yolo(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "coco")){ |
| | | run_coco(argc, argv); |
| | | } else if (0 == strcmp(argv[1], "dice")){ |