faster-rcnn环境配置
源代码:https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0
环境配置:pytorch1.10,python3.7,cuda10.0
细节:
1.安装依赖项:pip install -r requirements.txt
2.安装pycocotools:
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
3.pillow版本问题:卸载原pillow,换成6.2.2
4.scipy版本问题:卸载原scipy,换成1.2.0
5.在已训练的faster-rcnn模型上继续训练:使用--r参数
python trainval_net.py --dataset pascal_voc --net res101 --bs 1 --nw 4 --cuda --r True --checksession 1 --checkepoch 30 --checkpoint 5959
Pytorch模型保存与加载,并在加载的模型基础上继续训练 - 简书 (jianshu.com)
6.PR曲线,召回率
(23条消息) tf-faster-rcnn显示精度Precision和召回率Recall并保存PR曲线_折耳猫的橙汁儿的博客-CSDN博客
7.在测试集上测试
python test_net.py --dataset pascal_voc --net res101 --checksession 1 --checkepoch 30 --checkpoint 5959 --cuda
8.识别image内的图片
python demo.py --net vgg16 --checksession 1 --checkepoch 20 --checkpoint 1291 --cuda --load_dir models
参考教程:
(23条消息) 跑通Faster-RCNN Pytorch-1.0以及如何训练自己的数据集(详细到发抖)_HeiZee的博客-CSDN博客_faster rcnn训练自己的数据集
(23条消息) ubuntu+pytorch+faster rcnn训练自己的数据集_zxmyoung的博客-CSDN博客_使用pytorch版faster-rcnn训练自己数据集 主要

浙公网安备 33010602011771号