mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

:charmander:

Browse Source
This commit is contained in:
Joseph Redmon 2016-11-16 00:23:38 -08:00
parent fc9b867dd9
commit 5c70f1b6c8
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/darknet.c
View File

@ -12,6 +12,7 @@
#include "opencv2/highgui/highgui_c.h" #include "opencv2/highgui/highgui_c.h"
#endif #endif
extern void predict_classifier(char *datacfg, char *cfgfile, char *weightfile, char *filename, int top);
extern void run_voxel(int argc, char **argv); extern void run_voxel(int argc, char **argv);
extern void run_yolo(int argc, char **argv); extern void run_yolo(int argc, char **argv);
extern void run_detector(int argc, char **argv); extern void run_detector(int argc, char **argv);
@ -388,6 +389,8 @@ int main(int argc, char **argv)
run_vid_rnn(argc, argv); run_vid_rnn(argc, argv);
} else if (0 == strcmp(argv[1], "coco")){ } else if (0 == strcmp(argv[1], "coco")){
run_coco(argc, argv); run_coco(argc, argv);
} else if (0 == strcmp(argv[1], "classify")){
predict_classifier("cfg/imagenet1k.dataset", argv[2], argv[3], argv[4], 10);
} else if (0 == strcmp(argv[1], "classifier")){ } else if (0 == strcmp(argv[1], "classifier")){
run_classifier(argc, argv); run_classifier(argc, argv);
} else if (0 == strcmp(argv[1], "art")){ } else if (0 == strcmp(argv[1], "art")){