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 |
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训练自己数据集 主要
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾(3.3-3.9)
· AI 智能体引爆开源社区「GitHub 热点速览」
· 写一个简单的SQL生成工具