使用的源码https://github.com/endernewton/tf-faster-rcnn

前期准备工作可以参看该博主的第一到五步的步骤tf-faster-rcnn指南

我主要碰到的问题是使用kitti数据集训练是报错keyerror:car
原因应该是我的xml文件中有car,而我设定的类别中是Car 而没有car
我自己尝试了使用了labelimg标注了10张图片作为测试,label均为car后,验证了这个问题产生原因

还有一个问题是训练过程中total loss=nan
可以参考该博主的解决方案total loss=nan的解决方案

实际表示能解决问题

tf-faster-rcnn训练自己的数据集(学习记录)
最后附上训练200次的结果 用的是res101的预训练模型


今天训练了50000次kitti数据集后
tf-faster-rcnn训练自己的数据集(学习记录)
然后运行demo.py 参考该博主demo.py修改
效果不错

相关文章:

  • 2022-01-25
  • 2021-09-20
  • 2022-01-20
  • 2019-09-21
  • 2021-09-03
  • 2021-10-07
  • 2022-12-23
猜你喜欢
  • 2021-05-29
  • 2022-01-07
  • 2021-08-03
  • 2021-07-15
  • 2022-12-23
  • 2022-12-23
  • 2021-05-14
相关资源
相似解决方案