| | |
| | | void train_coco(char *cfgfile, char *weightfile) |
| | | { |
| | | //char *train_images = "/home/pjreddie/data/voc/test/train.txt"; |
| | | char *train_images = "/home/pjreddie/data/coco/train.txt"; |
| | | //char *train_images = "/home/pjreddie/data/coco/train.txt"; |
| | | char *train_images = "data/coco.trainval.txt"; |
| | | char *backup_directory = "/home/pjreddie/backup/"; |
| | | srand(time(0)); |
| | | data_seed = time(0); |
| | |
| | | void demo_coco(char *cfgfile, char *weightfile, float thresh, int cam_index, char *filename); |
| | | static void demo(char *cfgfile, char *weightfile, float thresh, int cam_index, char* filename) |
| | | { |
| | | #if defined(OPENCV) && defined(GPU) |
| | | #if defined(OPENCV) |
| | | demo_coco(cfgfile, weightfile, thresh, cam_index, filename); |
| | | #else |
| | | fprintf(stderr, "Need to compile with GPU and OpenCV for demo.\n"); |
| | | fprintf(stderr, "Need to compile with OpenCV for demo.\n"); |
| | | #endif |
| | | } |
| | | |