From 6103ecc4012ae8a93b96e808bc05f26cdef470d9 Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Mon, 9 Nov 2015 12:06:54 -0800 Subject: [PATCH] ready for push? --- Makefile | 4 ++-- cfg/yolo.cfg | 2 +- src/imagenet.c | 6 ------ src/yolo.c | 19 ++++--------------- 4 files changed, 7 insertions(+), 24 deletions(-) diff --git a/Makefile b/Makefile index 412e8264..44a193fc 100644 --- a/Makefile +++ b/Makefile @@ -1,5 +1,5 @@ -GPU=1 -OPENCV=1 +GPU=0 +OPENCV=0 DEBUG=0 ARCH= --gpu-architecture=compute_20 --gpu-code=compute_20 diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg index a652f466..8664831e 100644 --- a/cfg/yolo.cfg +++ b/cfg/yolo.cfg @@ -1,6 +1,6 @@ [net] batch=64 -subdivisions=2 +subdivisions=64 height=448 width=448 channels=3 diff --git a/src/imagenet.c b/src/imagenet.c index fa162519..dece9528 100644 --- a/src/imagenet.c +++ b/src/imagenet.c @@ -48,12 +48,6 @@ void train_imagenet(char *cfgfile, char *weightfile) pthread_join(load_thread, 0); train = buffer; - /* - image im = float_to_image(256, 256, 3, train.X.vals[114]); - show_image(im, "training"); - cvWaitKey(0); - */ - load_thread = load_data_in_thread(args); printf("Loaded: %lf seconds\n", sec(clock()-time)); time=clock(); diff --git a/src/yolo.c b/src/yolo.c index 77dae396..2abfa133 100644 --- a/src/yolo.c +++ b/src/yolo.c @@ -46,13 +46,7 @@ void draw_yolo(image im, int num, float thresh, box *boxes, float **probs, char 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 @@ void validate_yolo(char *cfgfile, char *weightfile) 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 @@ void validate_yolo_recall(char *cfgfile, char *weightfile) 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 @@ void test_yolo(char *cfgfile, char *weightfile, char *filename, float thresh) 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 *));