From 5e448c00f32e80e13eead72318047ce8368aac9a Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Mon, 05 Mar 2018 16:59:29 +0000 Subject: [PATCH] Added Focal loss. Use focal_loss=1 in the [region] layer in your cfg-file. alpha = 0.5, gamma = 2 --- src/data.h | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/src/data.h b/src/data.h index 7838681..e80beae 100644 --- a/src/data.h +++ b/src/data.h @@ -83,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); -- Gitblit v1.10.0