Macros | |
#define | NUMCHARS 37 |
Functions | |
void | blend_truth (float *new_truth, int boxes, int truth_size, float *old_truth) |
void | blend_truth_mosaic (float *new_truth, int boxes, int truth_size, float *old_truth, int w, int h, float cut_x, float cut_y, int i_mixup, int left_shift, int right_shift, int top_shift, int bot_shift, int net_w, int net_h, int mosaic_bound) |
data | concat_data (data d1, data d2) |
matrix | concat_matrix (matrix m1, matrix m2) |
void | correct_boxes (box_label *boxes, int n, float dx, float dy, float sx, float sy, int flip) |
void | fill_hierarchy (float *truth, int k, Darknet::Tree *hierarchy) |
int | fill_truth_detection (const char *path, int num_boxes, int truth_size, float *truth, int classes, int flip, float dx, float dy, float sx, float sy, int net_w, int net_h) |
data | get_data_part (data d, int part, int total) |
void | get_next_batch (data d, int n, int offset, float *X, float *y) |
list * | get_paths (const char *filename) |
char ** | get_random_paths (char **paths, int n, int m) |
char ** | get_random_paths_custom (char **paths, int n, int m, int contrastive) |
char ** | get_sequential_paths (char **paths, int n, int m, int mini_batch, int augment_speed, int contrastive) |
data | load_data_detection (int n, char **paths, int m, int w, int h, int c, int boxes, int truth_size, int classes, int use_flip, int use_gaussian_noise, int use_blur, int use_mixup, float jitter, float resize, float hue, float saturation, float exposure, int mini_batch, int track, int augment_speed, int letter_box, int mosaic_bound, int contrastive, int contrastive_jit_flip, int contrastive_color, int show_imgs) |
void | randomize_boxes (box_label *b, int n) |
box_label * | read_boxes (char *filename, int *n) |
#define NUMCHARS 37 |
void blend_truth | ( | float * | new_truth, |
int | boxes, | ||
int | truth_size, | ||
float * | old_truth | ||
) |
void blend_truth_mosaic | ( | float * | new_truth, |
int | boxes, | ||
int | truth_size, | ||
float * | old_truth, | ||
int | w, | ||
int | h, | ||
float | cut_x, | ||
float | cut_y, | ||
int | i_mixup, | ||
int | left_shift, | ||
int | right_shift, | ||
int | top_shift, | ||
int | bot_shift, | ||
int | net_w, | ||
int | net_h, | ||
int | mosaic_bound | ||
) |
void correct_boxes | ( | box_label * | boxes, |
int | n, | ||
float | dx, | ||
float | dy, | ||
float | sx, | ||
float | sy, | ||
int | flip | ||
) |
void fill_hierarchy | ( | float * | truth, |
int | k, | ||
Darknet::Tree * | hierarchy | ||
) |
int fill_truth_detection | ( | const char * | path, |
int | num_boxes, | ||
int | truth_size, | ||
float * | truth, | ||
int | classes, | ||
int | flip, | ||
float | dx, | ||
float | dy, | ||
float | sx, | ||
float | sy, | ||
int | net_w, | ||
int | net_h | ||
) |
void get_next_batch | ( | data | d, |
int | n, | ||
int | offset, | ||
float * | X, | ||
float * | y | ||
) |
list * get_paths | ( | const char * | filename | ) |
char ** get_random_paths | ( | char ** | paths, |
int | n, | ||
int | m | ||
) |
char ** get_random_paths_custom | ( | char ** | paths, |
int | n, | ||
int | m, | ||
int | contrastive | ||
) |
char ** get_sequential_paths | ( | char ** | paths, |
int | n, | ||
int | m, | ||
int | mini_batch, | ||
int | augment_speed, | ||
int | contrastive | ||
) |
data load_data_detection | ( | int | n, |
char ** | paths, | ||
int | m, | ||
int | w, | ||
int | h, | ||
int | c, | ||
int | boxes, | ||
int | truth_size, | ||
int | classes, | ||
int | use_flip, | ||
int | use_gaussian_noise, | ||
int | use_blur, | ||
int | use_mixup, | ||
float | jitter, | ||
float | resize, | ||
float | hue, | ||
float | saturation, | ||
float | exposure, | ||
int | mini_batch, | ||
int | track, | ||
int | augment_speed, | ||
int | letter_box, | ||
int | mosaic_bound, | ||
int | contrastive, | ||
int | contrastive_jit_flip, | ||
int | contrastive_color, | ||
int | show_imgs | ||
) |
void randomize_boxes | ( | box_label * | b, |
int | n | ||
) |
box_label * read_boxes | ( | char * | filename, |
int * | n | ||
) |