单发多框检测(SSD)

你的box_loss函数中先取所有损失值的平均数与参数做判断再对整个loss tensor做smooth,等于只是对小于或者大于参数的损失值做一边倒的smooth,没有精确到对每一个损失值的smooth化,故而得出的box_loss不会有太大变化,我的理解是先对于tensor内的每一个损失值做smooth再将其平均,这样可以做到蕴含每个损失值的smooth。我这边没有发现异常,附上我的代码,你可以试验一下:
smooth_l1_fn

感谢图片。

我遇到的一个就是做练习1的时候,修改了损失函数,但是最后在预测时没有重新加载预测图片,导致X是最后一个epoch的最后一个批量的特征,所以一直预测结果不对,以为是损失函数的问题,希望遇到类似问题的小伙伴可以看下是不是也有这个错误,对应代码