From f199fd3b6464e644566d76676c0b5f1824d26c4e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 17 Apr 2015 19:32:54 +0000
Subject: [PATCH] per image randomness in crop layer
---
src/detection.c | 13 ++++++++-----
1 files changed, 8 insertions(+), 5 deletions(-)
diff --git a/src/detection.c b/src/detection.c
index 1cf9ef0..cba3d18 100644
--- a/src/detection.c
+++ b/src/detection.c
@@ -57,6 +57,7 @@
void train_detection(char *cfgfile, char *weightfile)
{
srand(time(0));
+ data_seed = time(0);
int imgnet = 0;
char *base = basecfg(cfgfile);
printf("%s\n", base);
@@ -81,6 +82,8 @@
plist = get_paths("/home/pjreddie/data/imagenet/det.train.list");
}else{
plist = get_paths("/home/pjreddie/data/voc/trainall.txt");
+ //plist = get_paths("/home/pjreddie/data/coco/trainval.txt");
+ //plist = get_paths("/home/pjreddie/data/voc/all2007-2012.txt");
}
paths = (char **)list_to_array(plist);
pthread_t load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer);
@@ -93,9 +96,11 @@
load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer);
/*
- image im = float_to_image(im_dim, im_dim, 3, train.X.vals[114]);
- draw_detection(im, train.y.vals[114], 7);
-*/
+ image im = float_to_image(net.w, net.h, 3, train.X.vals[114]);
+ image copy = copy_image(im);
+ draw_detection(copy, train.y.vals[114], 7);
+ free_image(copy);
+ */
printf("Loaded: %lf seconds\n", sec(clock()-time));
time=clock();
@@ -198,8 +203,6 @@
fgets(filename, 256, stdin);
strtok(filename, "\n");
image im = load_image_color(filename, im_size, im_size);
- translate_image(im, -128);
- scale_image(im, 1/128.);
printf("%d %d %d\n", im.h, im.w, im.c);
float *X = im.data;
time=clock();
--
Gitblit v1.10.0