faster rcnn训练测试应该注意的问题

训练自己的数据集的时候,注意:

1.修改train.ptototxt

//TODO具体修改内容不表

2.修改test.prototxt

//TODO具体修改内容不表

3.修改代码中pascal_voc中的class

//具体修改内容根据自己的分类类别来更改

4.修改demo-my.py中的class,class顺序与pascal_voc.py中的类别标签的顺序要一致

//TODO注意类别顺序要一致,换成自己的数据测试


下面是最终要的地方!!!画重点!!!

5.测试注意事项

faster rcnn训练测试应该注意的问题_第1张图片

demo.py实际调用的是faster_rcnn_test.pt来测试的网络,所以这个地方也要相应的修改为对应的类别!!!

你可能感兴趣的:(faster,rcnn训练问题)