ubuntu下MASK-RCNN運作記錄

一，環境

1.ubuntu18.04 (tensorflow>=1.3.0)

2.tensorflow (tensorflow>=1.3.0)

3.keras>=2.0.8

二，下載下傳代碼

git clone https://github.com/matterport/Mask_RCNN
           

注意下載下傳時要下載下傳完全。

三，安裝Python coco API

git clone https://github.com/pdollar/coco.git
cd coco/PythonAPI
make
           

make後會生成檔案夾如圖

将編譯（make）後的pycocotools整個目錄拷貝到mask程式根目錄下

四，安裝依賴項

先cd到mask的根目錄下，有一個requirements.txt，再輸入：

pip3 install -r requirements.txt
           

五，修改demo

源檔案demo字尾為.ipynb，需要修改為.py字尾名。

1.打開Ipython Notebook

python -m IPython notebook
           

2.打開需要轉換的 .ipynb 檔案

需要修改的demo檔案在Mask_RCNN/samples檔案夾中

3.修改後儲存并複制到Mask_RCNN/samples檔案夾中

六，下載下傳權重檔案

百度網盤下載下傳：

連結：https://pan.baidu.com/s/1drKvfg 密碼：yer9

手動下載下傳：https://github.com/matterport/Mask_RCNN/releases/download/v2.0/mask_rcnn_coco.h5

下載下傳後放到根目錄下

七，修改demo.py

1.修改權重路徑

路徑修改為自己權重的儲存位置。

2.注釋掉：

3.運作demo.py檔案