１．下載代碼：

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

也可以暴力下載＜--直接download壓縮包--＞

2.　解壓完－－＞cd到faser-rcnn.pytorch文件夾中，－－＞再創建新的文件夾用來存放自己制作的數據集

３．cd ..返回到faster-rcnn.pytorch文件夾中－－＞然后在終端輸入

pip install -r requirements.txt

４．踩坑點！！！重要的事情感嘆號！！！

這一步有很多要修改的點

(1)首先針對錯誤 --can't import 'imread'－－＞需要將scipy的版本降低

pip install scipy==1.0

（２）cd到lib文件夾進行編譯

python setup.py build develop

（３）＊＊＊－－安裝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))

后面就是訓練啦＾＆＾

＃＃＃＃這太多大佬都出了博文，就不重復啦，也只是對踩坑的地方做個記錄＃＃＃＃

總結

以上是生活随笔為你收集整理的faster rcnn 训练自己的数据集－－－踩坑记录！！！的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。