使用的源码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的解决方案
实际表示能解决问题
最后附上训练200次的结果 用的是res101的预训练模型
今天训练了50000次kitti数据集后
然后运行demo.py 参考该博主demo.py修改
效果不错