From f996bd59a61338d8d51e2b19482d684f6dd04d0f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 23 Sep 2015 21:16:48 +0000
Subject: [PATCH] more writing fixes
---
src/data.h | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/data.h b/src/data.h
index 216ab0c..b91819f 100644
--- a/src/data.h
+++ b/src/data.h
@@ -26,7 +26,7 @@
} data;
typedef enum {
- CLASSIFICATION_DATA, DETECTION_DATA, CAPTCHA_DATA, REGION_DATA, IMAGE_DATA, COMPARE_DATA
+ CLASSIFICATION_DATA, DETECTION_DATA, CAPTCHA_DATA, REGION_DATA, IMAGE_DATA, COMPARE_DATA, WRITING_DATA
} data_type;
typedef struct load_args{
@@ -37,6 +37,8 @@
char **labels;
int h;
int w;
+ int out_w;
+ int out_h;
int nh;
int nw;
int num_boxes;
@@ -68,7 +70,7 @@
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 out_w, int out_h);
list *get_paths(char *filename);
char **get_labels(char *filename);
--
Gitblit v1.10.0