AlexeyAB
2018-02-08 cd8a3dcb4ca42f22ad8f46a95e00977c92be6bbd
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"
@@ -49,6 +53,7 @@
    int classes;
    int background;
    int scale;
   int small_object;
    float jitter;
    float angle;
    float aspect;
@@ -78,7 +83,7 @@
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_detection(int n, char **paths, int m, int w, int h, int boxes, int classes, float jitter, float hue, float saturation, float exposure, int small_object);
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);