single shot yolo training, separate file

This commit is contained in:
Joseph Redmon
2015-08-14 11:45:11 -07:00
parent e625b32e06
commit 2afa376bb3
4 changed files with 329 additions and 2 deletions

View File

@ -12,6 +12,7 @@
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);
@ -115,6 +116,8 @@ int main(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")){