Added param counters_per_class=100,200,... <for each class> to [yolo] and [Gaussian_yolo] to solve class-imbalance-problem

This commit is contained in:
AlexeyAB
2019-12-17 18:03:42 +03:00
parent 63396082d7
commit e33ecb785e
7 changed files with 45 additions and 11 deletions

View File

@ -261,6 +261,7 @@ struct layer {
float learning_rate_scale;
float clip;
int focal_loss;
float *classes_multipliers;
float label_smooth_eps;
int noloss;
int softmax;