（1）mask-rcnn之Demo運行（win10&ubuntu）

 

目錄

 

 

1.依賴

2.下載源碼(python3.4+)

3.下載訓練模型文件，並放在主目錄下面

4.運行demo

4.1 啓動jupyter-notebook

4.2 打開simple/demo.ipynb，修改一下代碼

5.運行demo

6.可能會出現的問題

---

1.依賴

安裝tensorflow和keras由於版本不兼容走了不少彎路。目前可使用一下版本：

 pip install tensorflow==1.9.0 --user

 pip install keras==2.0.8 --user

安裝pycocotools

 pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

2.下載源碼(python3.4+)

# 克隆源碼
git clone https://github.com/matterport/Mask_RCNN.git
# 進入項目目錄
cd Mask_RCNN
# 安裝依賴
pip3 install -r requirements.txt
# 安裝項目
python setup.py install

3.下載訓練模型文件，並放在主目錄下面

1. 在https://github.com/matterport/Mask_RCNN/releases下載mask_rcnn_coco.h5
2. 也可以在百度網盤下載：鏈接: https://pan.baidu.com/s/1XQfwuCrHUR9PwK28jXSuGg 提取碼: dhf7 

注意：如果沒有科學上網，在github下載文件可能不完整，會報“OSError: Unable to open file (truncated file: eof = 24614474, sblock->base_addr = 0, stored_eof = 257557808)：”的錯誤，我在這裏卡了很久，慚愧了！

4.運行demo

4.1 啓動jupyter-notebook

# 進入項目主目錄
cd Mask_RCNN

# 打開jupyter-notebook
jupyter-notebook

4.2 打開simple/demo.ipynb，修改一下代碼

模型文件名稱

修改這兩個變量，不然會報“len(images) must be equal to BATCH_SIZE”，這裏看了一下源碼：self.BATCH_SIZE = self.IMAGES_PER_GPU * self.GPU_COUNT

5.運行demo

點擊cell--運行全部,即可看到結果

6.可能會出現的問題

1. 缺各種包，缺什麼安裝就好。
2. tensorflow，keras中提示沒有moudle，按照步驟1安裝兼容版本的tensorflow，keras