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

(23条消息) if args.resume:从断点处开始继续训练模型——How to resume training?(Faster RCNN)_Yale曼陀罗的博客-CSDN博客_args.resume

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训练自己数据集         主要

posted @ 2022-04-07 00:52  云上飘啊飘  阅读(326)  评论(0)    收藏  举报