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

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))

后面就是训练啦^&^

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

你可能感兴趣的:(深度学习,人工智能,python)