首先需要下載 mask_rcnn 的源碼
但是從GitHub上下載源碼的速度非常慢,所以我們從碼雲上下載,這是GitHub的中國鏡像。
鏈接如下:https://gitee.com/mirrors/Mask_RCNN?_from=gitee_search
下載pycocotools
什么需要安裝pycocotools,經過看源碼發現,訓練coco數據集時用到了pycocotools這個模塊,如果不安裝會報錯無法正常運行。
到 https://github.com/philferriere/cocoapi 下載源碼,並進行解壓。以管理員身份打開 CMD 終端,並切換到 cocoapi\PythonAPI目錄。運行以下指令:
#install pycocotools locally
python setup.py build_ext --inplace
#install pycocotools to the Python site-packages
python setup.py build_ext install
下載預先訓練的COCO權重(mask_rcnn_coco.h5)
- 這個可以從GitHub下載
- 或者留下郵箱,我給你發過去
- 或者去CSDN的下載資源去下載
- 下載后放在 mask_rcnn 的根目錄
安裝
- CD 進入 mask_rcnn 項目根目錄
pip install -r requirements.txt
- 仍是在根目錄運行安裝程序
python3 setup.py install
(不報錯的話就安裝完成了,如果報錯可以根據錯誤提示,網絡搜索解決。python3 不行的話就用 python。
還要注意一點你使用哪個python環境安裝,后面運行的時候也要用此python環境運行MaskRCNN)
演示
- cd 到 Mask_RCNN\samples目錄
- 用安裝Mask RCNN的python環境打開 jupyter notebook,命令行,或shell運行:
jupyter notebook
- 打開 samples/demo.ipynb 文件,運行所有cell,這樣就可以運行demo了。
- 可以更換images目錄里面的圖片,用自己的圖片進行測試