在跑 Faster-RCNN 的代码的时候,可能手残想更改官方的Pascal VOC数据集,改成跑自己的数据集。
但是可能会出现这样或者类似的warning:
RuntimeWarning: invalid value encountered in true_divide
RuntimeWarning: invalid value encountered in greater_equal
...
接着可能就会出现loss = nan的情况,且一直不见好转
通过一步步调试和网上搜集,发现可能的原因:
就是这篇博客给出的原因,具体讨论也可以参照rbg大佬github官方issue。
由于我们更换了数据集,所以在制作自己的数据集的annotation时候,可能会出现bbox的宽w和高h为0的情况,这样导致在代码中计算偏移的时候,或者
为负无穷大。