Mask R-CNN執行demo記錄
阿新 • • 發佈:2018-12-18
一、配置環境:
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 )
二、安裝
-
安裝依賴關係
pip3 install -r requirements.txt
-
執行安裝程式
python3 setup.py install
- 下載訓練好的權重:mask_rcnn_coco.h5,地址:releases
- 安裝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)