From 0df9d25c46e39fa5f532c023261784fdcdd5d25d Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Jun 2015 22:51:17 +0000
Subject: [PATCH] new cfg files for classification
---
src/darknet.c | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/darknet.c b/src/darknet.c
index 0679ffe..0a705da 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -77,6 +77,7 @@
void rgbgr_filters(convolutional_layer l);
void rgbgr_net(char *cfgfile, char *weightfile, char *outfile)
{
+ gpu_index = -1;
network net = parse_network_cfg(cfgfile);
if(weightfile){
load_weights(&net, weightfile);
@@ -106,7 +107,7 @@
int main(int argc, char **argv)
{
- //test_resize("data/cat.png");
+ //test_resize("data/bad.jpg");
//test_box();
//test_convolutional_layer();
if(argc < 2){
@@ -142,6 +143,8 @@
partial(argv[2], argv[3], argv[4], atoi(argv[5]));
} else if (0 == strcmp(argv[1], "visualize")){
visualize(argv[2], (argc > 3) ? argv[3] : 0);
+ } else if (0 == strcmp(argv[1], "imtest")){
+ test_resize(argv[2]);
} else {
fprintf(stderr, "Not an option: %s\n", argv[1]);
}
--
Gitblit v1.10.0