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 | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index 989bf6f..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);
@@ -127,7 +128,7 @@
network net = parse_network_cfg(cfgfile);
int oldn = net.layers[net.n - 2].n;
int c = net.layers[net.n - 2].c;
- net.layers[net.n - 2].n = 7879;
+ net.layers[net.n - 2].n = 9372;
net.layers[net.n - 2].biases += 5;
net.layers[net.n - 2].weights += 5*c;
if(weightfile){
@@ -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