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:

1
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参数

1
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 @   云上飘啊飘  阅读(298)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾(3.3-3.9)
· AI 智能体引爆开源社区「GitHub 热点速览」
· 写一个简单的SQL生成工具
点击右上角即可分享
微信分享提示