writing stuff

cfg/writing.cfg

@@ -0,0 +1,49 @@
+[net]
+batch=64
+subdivisions=1
+height=256
+width=256
+channels=3
+learning_rate=0.00001
+momentum=0.9
+decay=0.0005
+seen=0
+
+[crop]
+crop_height=256
+crop_width=256
+flip=0
+angle=0
+saturation=1
+exposure=1
+
+[convolutional]
+filters=32
+size=3
+stride=1
+pad=1
+activation=ramp
+
+[convolutional]
+filters=32
+size=3
+stride=1
+pad=1
+activation=ramp
+
+[convolutional]
+filters=32
+size=3
+stride=1
+pad=1
+activation=ramp
+
+[convolutional]
+filters=1
+size=5
+stride=1
+pad=1
+activation=logistic
+
+[cost]
+

src/data.c

@@ -54,7 +54,12 @@ matrix load_image_paths_gray(char **paths, int n, int w, int h)
     X.cols = 0;
 
     for(i = 0; i < n; ++i){
-        image im = load_image(paths[i], w, h, 1);
+        image im = load_image(paths[i], w, h, 3);
+
+        image gray = grayscale_image(im);
+        free_image(im);
+        im = gray;
+
         X.vals[i] = im.data;
         X.cols = im.h*im.w*im.c;
     }
@@ -571,14 +576,14 @@ pthread_t load_data_in_thread(load_args args)
     return thread;
 }
 
-data load_data_writing(char **paths, int n, int m, int w, int h)
+data load_data_writing(char **paths, int n, int m, int w, int h, int downsample)
 {
     if(m) paths = get_random_paths(paths, n, m);
     char **replace_paths = find_replace_paths(paths, n, ".png", "-label.png");
     data d;
     d.shallow = 0;
     d.X = load_image_paths(paths, n, w, h);
-    d.y = load_image_paths_gray(replace_paths, n, w/8, h/8);
+    d.y = load_image_paths_gray(replace_paths, n, w/downsample, h/downsample);
     if(m) free(paths);
     int i;
     for(i = 0; i < n; ++i) free(replace_paths[i]);

src/data.h

@@ -68,7 +68,7 @@ box_label *read_boxes(char *filename, int *n);
 data load_cifar10_data(char *filename);
 data load_all_cifar10();
 
-data load_data_writing(char **paths, int n, int m, int w, int h);
+data load_data_writing(char **paths, int n, int m, int w, int h, int downsample);
 
 list *get_paths(char *filename);
 char **get_labels(char *filename);