Added Focal loss. Use focal_loss=1 in the [region] layer in your cfg-file.

alpha = 0.5, gamma = 2
This commit is contained in:
AlexeyAB
2018-03-05 19:59:29 +03:00
parent f0bc634a74
commit 5e448c00f3
3 changed files with 31 additions and 9 deletions

View File

@ -85,6 +85,7 @@ struct layer{
float exposure;
float shift;
float ratio;
int focal_loss;
int softmax;
int classes;
int coords;