最近在学习计算机视觉,记录一下一个小白学习faster rcnn 的过程。
复现代码链接:https://github.com/ruotianluo/pytorch-faster-rcnn
复现环境:ubuntu18.04+pytorch1.3,GPU使用1080Ti。
一、利用预训练模型的测试图片
1、首先按照链接的要求,下载代码。
git clone https://github.com/ruotianluo/pytorch-faster-rcnn.git
2、下载数据集
我是在VOC2012数据集上进行测试,已经提前下载了VOC2012数据集。
然后在根目录下创建一个data文件夹,然后建立软链接来指向数据集的位置,方法如下:
cd $FRCN_ROOT/data
ln -s $VOCdevkit VOCdevk