Pytorch_Retinaface icon indicating copy to clipboard operation
Pytorch_Retinaface copied to clipboard

Loss balancing parameters and normalization

Open casperthuis opened this issue 5 years ago • 2 comments
trafficstars

Hi @biubug6,

First off, thank you for your work and for sharing. I had a question about your loss definitions.

The loss defined in the paper L_cls(pi, pi^) + lamda_1 * pi^ * L_box(ti, ti*) + lamda_2 * pi^* * L_pts(li, li*) + lamda_3 * pi^* * L_pixel. The loss contains 3 loss balancing parameters, which the paper say they set to 0.25, 0.1 and 0.01 respectively.

However, when looking in the code I only see one loss balancing parameter for the localization loss. Furthermore, I do not see the loss balancing happening in the multibox loss module. And do not see the pixel loss also. Is there a reason for this or will it be added later?

The second question is about the normalization of the loss inside the mutlibox loss module. The loss for the loss classes is normalized the same as the localizations, normally it is better to normalize the classes by all boxes and the localization by the positives ones. is there also a reason for this?

Thanks in advance.

Best,

Casper Thuis

casperthuis avatar Dec 05 '19 13:12 casperthuis

I have the same questions,

dianxin556 avatar Dec 30 '19 09:12 dianxin556

I have the same questions,too

a1391651300 avatar Feb 01 '21 08:02 a1391651300