一、 faster rcnn环境搭建
下载源码 https://github.com/rbgirshick/py-faster-rcnn python版本
https://github.com/ShaoqingRen/faster_rcnn Matlab版本
本文介绍matlab版本的配置过程:
1、 在Faster_rcnn-master目录下添加bin文件
2、 在Faster_rcnn-master目录下添加models文件
3、 在external/caffe目录下添加MATLAB文件
4、 添加datasets文件夹,文件夹内为 数据集,可以使用VOC2007或者自己的数据集
5、 如果想直接测试,不需要添加datasets文件夹在Faster_rcnn-master目录下添加output文件,文件为训练好的模型,就可以直接运行experiments/script_faster_rcnn_demo.m,注意:运行时目录应在主目录Faster_rcnn-master下
6、 如果训练VOC2007数据,只需下载VOC2007的数据集,存放在datasets文件夹中,如果训练自己的数据集,则需要把自己的数据集做成和VOC2007相同格式的。
二、 制作数据集
我的数据集的图片格式是png,标记内容存储在txt中,写一个txt到xml格式转换的数据接口。
三、 细节修改
1、 datasets/VOCdevkit2007/VOCcode/VOCinit.m
Faster RCNN 训练自己的数据集
最新推荐文章于 2024-05-06 15:44:10 发布