Pytorch_Retinaface
Pytorch_Retinaface copied to clipboard
Loss balancing parameters and normalization
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
I have the same questions,
I have the same questions,too