1. 程式人生 > >Mask R-CNN執行demo記錄

Mask R-CNN執行demo記錄

一、配置環境:

Mask R-CNN是基於Python3,Keras,TensorFlow。

Python 3.4+ TensorFlow 1.3+ Keras 2.0.8+ Jupyter Notebook Numpy, skimage, scipy, Pillow, cython, h5py

(我使用的是Ubuntu系統,安裝Anaconda3。)

一個坑:一開始使用的keras版本是2.2.0,結果發現太高了,沒有load_weights_from_hdf5_group_by_name,換成了2.1.6版本,pip install keras==2.1.6 )

二、安裝

  1. 安裝依賴關係

    pip3 install -r requirements.txt
  2. 執行安裝程式

    python3 setup.py install
  3. 下載訓練好的權重:mask_rcnn_coco.h5,地址:releases
  4. 安裝pycocotools:make生成對應的檔案,make之後將生成的pycocotools資料夾複製到samples中的coco資料夾。 

(1)Linux地址:https://github.com/waleedka/coco (For Python, run "make" under coco/PythonAPI)

You must have the Visual C++ 2015 build tools on your path (see the repo for additional details)

三、開啟jupyter notebook執行demo.ipynb