Pytorch版本的SSD_Multibox损失函数

multibox_loss（损失函数）

原文：https://blog.csdn.net/goodxin_ie/article/details/89675756

 

补充 负样本选取源码：

使用了 log_sum_exp函数对负样本置信度编码，目的是防止 一开始网络的predictions元素过大，产生溢出，比如exp(1000)，会生成 inf ，即导致overflow，溢出程序会报错，

下面巧妙用了两个 sort 函数的输出，

最后使用排序得到neg输出；

 

正负样本的选取排序很精巧，感叹作者的代码逻辑功底。