美文网首页
用Caltech数据训练Faster-RCNN

用Caltech数据训练Faster-RCNN

作者: 韦德爱老詹 | 来源:发表于2018-01-18 21:23 被阅读0次

前面文章用Caltech数据制作成VOC形式,但在实际训练的时候仍然有挺多问题,所需文件如下。



主要是Annotations的问题,有如下几点需要注意:

  • 1 Annotations中要包含所有图片的信息,即使对应图片中没有行人数据,但仍需要xml文件,即有多少张图片就有多少xml。


  • 2 注意box的边界大小,Xmin,Xmax,Ymin,Ymax都需要注意,不要越界,否则会报错(可以在生成xml代码中就对这几个参数进行限幅),否则assert (boxes[:, 2] >= boxes[:, 0]).all() 会报错。
    数据集整理完毕,接下来便是修改参数:
  • 1.lib/networks/中VGGnet_test.py和VGGnet_train.py中n_classes = 2
  • 2.lib/datasets/pascal_voc.pyself._classes = ('__background__', 'person')
    和tools/demo.py中CLASSES = ('__background__','person')
  • 3.pascal_voc.py中self._image_ext = '.png'的修改。
    搞了快一个星期,终于完工。

相关文章

网友评论

      本文标题:用Caltech数据训练Faster-RCNN

      本文链接:https://www.haomeiwen.com/subject/nocgoxtx.html