yolov3检测小物体loss为nan

问题:图片大小为1024x1024,第一个batch跑完的时候很正常,到了第二个batch以及之后的都是nan
可能的原因:

  1. 图像本身大小3307X2340,被resize成1024X1024,object发生了形变,本身就是小物体,resize之后就更不容易识别
  2. 我使用的权重文件是yolov3.cfg,说明这个权重对应的图片与我的图像差异太大
  3. 检查一系列参数

暂时想到这些,会来补充并填坑!

2020年9月5更新:
我最终loss为nan的原因是聚类那边写错代码了,我修改之后效果很好!!

猜你喜欢

转载自blog.csdn.net/weixin_43868576/article/details/108398072
今日推荐