OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed a bug in counters_per_class=

This commit is contained in:
AlexeyAB 2019-12-18 03:53:40 +03:00
parent e33ecb785e
commit e43a1c424d
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/data.c
View File

@ -1545,7 +1545,7 @@ data load_data_augment(char **paths, int n, int m, char **labels, int k, tree *h
int i, j;
for (i = 0; i < d2.X.rows; ++i) {
if (mixup == 4) mixup = rand_int(2, 3); // alternate MixUp and CutMix
if (mixup == 4) mixup = rand_int(2, 3); // alternate CutMix and Mosaic
// MixUp -----------------------------------
if (mixup == 1) {

2
src/gaussian_yolo_layer.c
View File

@ -341,7 +341,7 @@ void delta_gaussian_yolo_class(float *output, float *delta, int index, int class
}
for(n = 0; n < classes; ++n){
delta[index + stride*n] = ((n == class_id) ? (1 - label_smooth_eps) : (0 + label_smooth_eps/classes)) - output[index + stride*n];
if (classes_multipliers) delta[index + stride*class_id] *= classes_multipliers[class_id];
if (classes_multipliers && n == class_id) delta[index + stride*class_id] *= classes_multipliers[class_id];
if(n == class_id && avg_cat) *avg_cat += output[index + stride*n];
}
}

2
src/yolo_layer.c
View File

@ -250,7 +250,7 @@ void delta_yolo_class(float *output, float *delta, int index, int class_id, int
// default
for (n = 0; n < classes; ++n) {
delta[index + stride*n] = ((n == class_id) ? (1 - label_smooth_eps) : (0 + label_smooth_eps/classes)) - output[index + stride*n];
if (classes_multipliers) delta[index + stride*class_id] *= classes_multipliers[class_id];
if (classes_multipliers && n == class_id) delta[index + stride*class_id] *= classes_multipliers[class_id];
if (n == class_id && avg_cat) *avg_cat += output[index + stride*n];
}
}