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))
后面就是训练啦^&^
####这太多大佬都出了博文,就不重复啦,也只是对踩坑的地方做个记录####