blame | history | patch | commit | commitdiff | ignore whitespace
IlyaOvodov
2018-06-04 f0abcfa02b2094396f955c743f7f11fcdb2e3d13 
 src/data.h


@@ -2,6 +2,10 @@


#define DATA_H


#include <pthread.h>




#if defined(_MSC_VER) && _MSC_VER < 1900


   #define inline __inline


#endif




#include "matrix.h"


#include "list.h"


#include "image.h"


@@ -28,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{


@@ -40,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;


@@ -49,7 +54,9 @@


    int classes;


    int background;


    int scale;


   int small_object;


    float jitter;


    int flip;


    float angle;


    float aspect;


    float saturation;


@@ -78,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);