训练显示nan表示主要损失和信心,0表示确认

时间:2019-03-20 08:22:19

标签: image-processing computer-vision training-data chainercv

我在训练时使用SSD512(图像网络预训练模型)和Faster_R-CNN(预训练),损失和置信度显示nan和验证为0。 我使用了与SSD300(图像网络预训练模型)相同的数据集和技术,该模型在训练[Basketball-ChainerCV](https://github.com/atom2k17/Basketball-ChainerCV/blob/master/basketballproject.py)后可以很好地给出良好的预测。

SSD512 training image link

在训练开始之前训练Faster R-CNN时,在第一组时间段的结果之前会显示以下内容:

/usr/local/lib/python3.6/dist- 
packages/chainercv/links/model/faster_rcnn/utils/loc2bbox.py:65: 
RuntimeWarning: overflow encountered in exp
  h = xp.exp(dh) * src_height[:, xp.newaxis]
/usr/local/lib/python3.6/dist- 
packages/chainercv/links/model/faster_rcnn/utils/loc2bbox.py:65: 
RuntimeWarning: overflow encountered in multiply
  h = xp.exp(dh) * src_height[:, xp.newaxis]
/usr/local/lib/python3.6/dist- 
packages/chainercv/links/model/faster_rcnn/utils/loc2bbox.py:66: 
RuntimeWarning: overflow encountered in exp
  w = xp.exp(dw) * src_width[:, xp.newaxis]
/usr/local/lib/python3.6/dist- 
packages/chainercv/links/model/faster_rcnn/utils/loc2bbox.py:66: 
RuntimeWarning: overflow encountered in multiply
  w = xp.exp(dw) * src_width[:, xp.newaxis]
/usr/local/lib/python3.6/dist- 
packages/chainercv/links/model/faster_rcnn/utils/proposal_creator.py:126: 
RuntimeWarning: invalid value encountered in greater_equal 

Faster_R-CNN training image link

我尝试过的事情:

  1. 提高学习率
  2. 减少batch_size
  3. 在文本文件中删除图像,注释和内容的图像 边框小于图片总大小的1%

注意:一切都可以与SSD300完美配合,而问题在于SSD512和Faster RCNN模型。

问题背后的问题是什么?谁能指出如何处理此类问题?

0 个答案:

没有答案