From 5c70f1b6c86d43f9f42e8061381722afb319ed5d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Nov 2016 08:23:38 +0000
Subject: [PATCH] :charmander:
---
src/darknet.c | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index 4419107..9657882 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 @@
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")){
--
Gitblit v1.10.0