From 5c70f1b6c86d43f9f42e8061381722afb319ed5d Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Wed, 16 Nov 2016 00:23:38 -0800 Subject: [PATCH] :charmander: --- src/darknet.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/darknet.c b/src/darknet.c index 4419107a..96578822 100644 --- 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")){