Added scale_x_y param to [yolo]-layer (for sigmoid).

In previous commit is used GIoU(generalized-iou) from  https://github.com/generalized-iou/g-darknet
This commit is contained in:
AlexeyAB
2019-06-04 21:20:59 +03:00
parent 6e13527f06
commit eac26226a7
8 changed files with 30 additions and 4 deletions

View File

@ -314,6 +314,7 @@ struct layer {
float *weights;
float *weight_updates;
float scale_x_y;
float iou_normalizer;
float cls_normalizer;
int iou_loss;