From 6103ecc4012ae8a93b96e808bc05f26cdef470d9 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 09 Nov 2015 20:06:54 +0000
Subject: [PATCH] ready for push?
---
src/yolo.c | 19 ++++---------------
1 files changed, 4 insertions(+), 15 deletions(-)
diff --git a/src/yolo.c b/src/yolo.c
index 77dae39..2abfa13 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -46,13 +46,7 @@
void train_yolo(char *cfgfile, char *weightfile)
{
- //char *train_images = "/home/pjreddie/data/voc/person_detection/2010_person.txt";
- //char *train_images = "/home/pjreddie/data/people-art/train.txt";
- //char *train_images = "/home/pjreddie/data/voc/test/2012_trainval.txt";
- //char *train_images = "/home/pjreddie/data/voc/test/2010_trainval.txt";
- char *train_images = "/home/pjreddie/data/voc/test/train.txt";
- //char *train_images = "/home/pjreddie/data/voc/test/train_all.txt";
- //char *train_images = "/home/pjreddie/data/voc/test/2007_trainval.txt";
+ char *train_images = "data/voc.0712.trainval";
char *backup_directory = "/home/pjreddie/backup/";
srand(time(0));
data_seed = time(0);
@@ -189,12 +183,8 @@
srand(time(0));
char *base = "results/comp4_det_test_";
- //base = "/home/pjreddie/comp4_det_test_";
- //list *plist = get_paths("/home/pjreddie/data/people-art/test.txt");
- //list *plist = get_paths("/home/pjreddie/data/cubist/test.txt");
-
- list *plist = get_paths("/home/pjreddie/data/voc/test/2007_test.txt");
- //list *plist = get_paths("/home/pjreddie/data/voc/test_2012.txt");
+ //list *plist = get_paths("data/voc.2007.test");
+ list *plist = get_paths("data/voc.2012.test");
char **paths = (char **)list_to_array(plist);
layer l = net.layers[net.n-1];
@@ -282,7 +272,7 @@
srand(time(0));
char *base = "results/comp4_det_test_";
- list *plist = get_paths("/home/pjreddie/data/voc/test/2007_test.txt");
+ list *plist = get_paths("data/voc.2007.test");
char **paths = (char **)list_to_array(plist);
layer l = net.layers[net.n-1];
@@ -373,7 +363,6 @@
char *input = buff;
int j;
float nms=.5;
- printf("%d %d %d", l.side, l.n, l.classes);
box *boxes = calloc(l.side*l.side*l.n, sizeof(box));
float **probs = calloc(l.side*l.side*l.n, sizeof(float *));
for(j = 0; j < l.side*l.side*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *));
--
Gitblit v1.10.0