faster rcnn

环境配置

python3.8

pytorch1.10.1

RTX2080Ti

数据集

VOC数据集下载地址：[https://pan.baidu.com/s/1-1Ej6dayrx3g0iAA88uY5A]

提取码：ph32

预训练网络

网络参数下载地址：[https://pan.baidu.com/s/1S6wG8sEXBeoSec95NZxmlQ]

提取码：8mgp

下载好的模型放到model_data目录下

训练

python voc_annotation.py

python trian.py

预测

在frcnn.py中修改model_path对应到自己训练的文件

python predict.py

然后输入预测图像地址

Reference

[https://github.com/bubbliiiing/faster-rcnn-pytorch]