:charmander:

minhtcai · Nov 16, 2016 · 5c70f1b · 5c70f1b
1 parent fc9b867
commit 5c70f1b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/darknet.c b/src/darknet.c
@@ -12,6 +12,7 @@
 #include "opencv2/highgui/highgui_c.h"
 #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_yolo(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);
     } else if (0 == strcmp(argv[1], "coco")){
         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")){
         run_classifier(argc, argv);
     } else if (0 == strcmp(argv[1], "art")){