faster rcnn 训练自己的数据集---踩坑记录!!!

本文档详细记录了在安装和配置Faster R-CNN在PyTorch环境中的步骤,包括从源码下载、依赖库的安装、Scipy版本降级、COCO API的安装及解决相关错误。特别指出了解决'import imread'错误、'_mask.so: undefined symbol:_Py_ZeroStruct'问题和'yaml.load()'缺少参数的问题。最后提醒读者训练部分可参考其他资源。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1.下载代码:

git clone https://github.com/jwyang/faster-rcnn.pytorch.git

也可以暴力下载<--直接download压缩包-->

2. 解压完-->cd到faser-rcnn.pytorch文件夹中,-->再创建新的文件夹用来存放自己制作的数据集

3.cd ..返回到faster-rcnn.pytorch文件夹中-->然后在终端输入

pip install -r requirements.txt 

4.踩坑点!!!重要的事情感叹号!!!

这一步有很多要修改的点

(1)首先针对错误 --can't import 'imread'-->需要将scipy的版本降低

pip install scipy==1.0

(2)cd到lib文件夹进行编译

python setup.py build develop

(3)***--安装CoCo API--***

针对错误 

ImportError: cannot import name '_mask'和

ModuleNotFoundError: No module named 'pycocotools._mask'-->需要下载cocoapi

git clone https://github.com/cocodataset/cocoapi.git

随便你放在什么位置,但还是建议放在lib里面,方便看,-->解压后cd到PYTHONAPI文件夹中-->如果是在ubuntu平台需要给一个权限,即加了前面sudo指令

sudo python setup.py install

还有一个坑...

_mask.so : undefined symbol: _Py_ZeroStruct

这时候需要-->

如果前面给了权限且不是另外开一个终端输入命令的情况下,其实也不需要sudo的

sudo python setup.py build_ext --inplace
sudo python setup.py build_ext install
sudo make

最最最后一个坑

TypeError: load() missing 1 required positional argument: 'Loader'

打开/lib/model/utils/config.py文件--> ctrl+f 搜索 

yaml_cfg = edict(yaml.load(f))

并将此改为

yaml_cfg = edict(yaml.load(f, Loader=yaml.FullLoader))

后面就是训练啦^&^

####这太多大佬都出了博文,就不重复啦,也只是对踩坑的地方做个记录####

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值