mask_rcnn(Keras+TensorFlow)環境搭建_新手向(畢業設計使用,親測可用)


首先需要下載 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 的根目錄

安裝

  1. CD 進入 mask_rcnn 項目根目錄
pip install -r requirements.txt
  1. 仍是在根目錄運行安裝程序
python3 setup.py install

(不報錯的話就安裝完成了,如果報錯可以根據錯誤提示,網絡搜索解決。python3 不行的話就用 python。
還要注意一點你使用哪個python環境安裝,后面運行的時候也要用此python環境運行MaskRCNN)

演示

  1. cd 到 Mask_RCNN\samples目錄
  2. 用安裝Mask RCNN的python環境打開 jupyter notebook,命令行,或shell運行:jupyter notebook
  3. 打開 samples/demo.ipynb 文件,運行所有cell,這樣就可以運行demo了。
  4. 可以更換images目錄里面的圖片,用自己的圖片進行測試

DONE~

有問題歡迎留言評論


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM