Faster RCNN 模型训练及检测

最新推荐文章于 2024-04-21 15:46:09 发布

Citroning

最新推荐文章于 2024-04-21 15:46:09 发布

阅读量1.6k

点赞数

分类专栏： Caffe CNN 文章标签：深度学习

本文链接：https://blog.csdn.net/qq_34335194/article/details/52346522

版权

本文详述了使用Faster RCNN进行目标检测的全过程，包括模型训练、数据集准备、模型保存和检测执行。通过VGG16/faster_rcnn_alt_opt网络，针对PASCAL VOC2007数据集进行训练，提供了训练配置修改、模型保存与加载、以及检测演示的步骤。

摘要由CSDN通过智能技术生成

本文针对faster rcnn下的模型训练及检测。所用网络VGG16/faster_rcnn_alt_opt, 数据集pascal voc2007 .
(友情提示：若要修改某文件里的信息，最好将原文件重命名备份，以免修改后出现问题导致无法复原。)

一更改类别（选做）
1 打开主目录（YOURS/Downloads/py-faster-rcnn，下同)下 models/pascal_voc/VGG16/faster_rcnn_alt_opt 文件夹里的若干文件，更改类别数。详细步骤可参考http://blog.csdn.net/u013078356/article/details/50987845#reply

2 打开 lib/datasets/pascal_voc.py, 修改self._classes 中的标签信息。此标签名称应与样本.xml文件中的标签名称一致。

二准备数据集（选做）
1 在 data/VOCdevkit2007/VOC2007 文件夹中，将原有的文件重命名用以备份。然后将含有样本标记信息的.xml文件放入新建的Annotations文件夹，将含有trainval和test样本名称（写成一列，名称后加一个空格，不带后缀，不留空行）的trainval.txt和test.txt文件放入新建的ImageSets/Main文件夹，将所有的图片放入新建的JPEGImages文件夹。

2 以上