blame | history | patch | commit | commitdiff | ignore whitespace
Tino Hager
2018-06-27 bff7644f31501fb8dd547e032e5ef6de67cf673e 
 src/data.h


@@ -32,7 +32,7 @@


} data;




typedef enum {


    CLASSIFICATION_DATA, DETECTION_DATA, CAPTCHA_DATA, REGION_DATA, IMAGE_DATA, COMPARE_DATA, WRITING_DATA, SWAG_DATA, TAG_DATA, OLD_CLASSIFICATION_DATA, STUDY_DATA, DET_DATA, SUPER_DATA


    CLASSIFICATION_DATA, DETECTION_DATA, CAPTCHA_DATA, REGION_DATA, IMAGE_DATA, LETTERBOX_DATA, COMPARE_DATA, WRITING_DATA, SWAG_DATA, TAG_DATA, OLD_CLASSIFICATION_DATA, STUDY_DATA, DET_DATA, SUPER_DATA


} data_type;




typedef struct load_args{


@@ -44,7 +44,8 @@


    char **labels;


    int h;


    int w;


    int out_w;


   int c; // color depth


   int out_w;


    int out_h;


    int nh;


    int nw;


@@ -53,7 +54,9 @@


    int classes;


    int background;


    int scale;


   int small_object;


    float jitter;


    int flip;


    float angle;


    float aspect;


    float saturation;


@@ -82,11 +85,11 @@


data load_data_captcha(char **paths, int n, int m, int k, int w, int h);


data load_data_captcha_encode(char **paths, int n, int m, int w, int h);


data load_data_old(char **paths, int n, int m, char **labels, int k, int w, int h);


data load_data_detection(int n, char **paths, int m, int w, int h, int boxes, int classes, float jitter, float hue, float saturation, float exposure);


data load_data_tag(char **paths, int n, int m, int k, int min, int max, int size, float angle, float aspect, float hue, float saturation, float exposure);


matrix load_image_augment_paths(char **paths, int n, int min, int max, int size, float angle, float aspect, float hue, float saturation, float exposure);


data load_data_detection(int n, char **paths, int m, int w, int h, int c, int boxes, int classes, int use_flip, float jitter, float hue, float saturation, float exposure, int small_object);


data load_data_tag(char **paths, int n, int m, int k, int use_flip, int min, int max, int size, float angle, float aspect, float hue, float saturation, float exposure);


matrix load_image_augment_paths(char **paths, int n, int use_flip, int min, int max, int size, float angle, float aspect, float hue, float saturation, float exposure);


data load_data_super(char **paths, int n, int m, int w, int h, int scale);


data load_data_augment(char **paths, int n, int m, char **labels, int k, tree *hierarchy, int min, int max, int size, float angle, float aspect, float hue, float saturation, float exposure);


data load_data_augment(char **paths, int n, int m, char **labels, int k, tree *hierarchy, int use_flip, int min, int max, int size, float angle, float aspect, float hue, float saturation, float exposure);


data load_go(char *filename);




box_label *read_boxes(char *filename, int *n);


@@ -97,6 +100,7 @@




list *get_paths(char *filename);


char **get_labels(char *filename);


char **get_labels_custom(char *filename, int *size);


void get_random_batch(data d, int n, float *X, float *y);


data get_data_part(data d, int part, int total);


data get_random_data(data d, int num);