blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2016-09-25 d05f87802a5edb569dd852117069f0a398ce9653 
 src/data.h


@@ -6,8 +6,6 @@


#include "list.h"


#include "image.h"




extern unsigned int data_seed;




static inline float distance_from_edge(int x, int max)


{


    int dx = (max/2) - x;


@@ -23,7 +21,6 @@


    int w, h;


    matrix X;


    matrix y;


    int *indexes;


    int shallow;


    int *num_boxes;


    box **boxes;


@@ -34,6 +31,7 @@


} data_type;




typedef struct load_args{


    int threads;


    char **paths;


    char *path;


    int n;


@@ -52,6 +50,7 @@


    int scale;


    float jitter;


    float angle;


    float aspect;


    float saturation;


    float exposure;


    float hue;


@@ -69,18 +68,19 @@




void free_data(data d);




pthread_t load_data(load_args args);




pthread_t load_data_in_thread(load_args args);




void print_letters(float *pred, int n);


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(char **paths, int n, int m, char **labels, int k, 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 hue, float saturation, float exposure);


matrix load_image_augment_paths(char **paths, int n, int min, int max, int size, float angle, 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_super(char **paths, int n, int m, int w, int h, int scale);


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


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


data load_data_augment(char **paths, int n, int m, char **labels, int k, 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);


@@ -92,6 +92,7 @@


list *get_paths(char *filename);


char **get_labels(char *filename);


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);


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


data load_categorical_data_csv(char *filename, int target, int k);