1. 程式人生 > >Tensorflow + Keras 執行Mask-RCNN

Tensorflow + Keras 執行Mask-RCNN

實驗環境:Ubuntu + tensorflow1.4 + cuda8.0 + python3.5 + Keras2.1

但這個repo已經較長時間不維護了,效果也不是很理想。於是選用了github上star數最多的MaskRCNN專案,在keras和tensorflow上再次進行實驗。

相關的部落格資料已經較為豐富,配置過程中主要參考了:

這裡對配置中遇到的問題做一下記錄。

1、關於notebook

一開始有個地方開沒裝完就打開了notebook,然後測試時報錯,後來發現是notebook環境沒有更新。需要在命令列裡重啟notebook,再次從頭執行demo檔案,即可順利載入模組。

2、關於pycocotools的安裝

根據參考部落格內容,如果是Ubuntu系統,pythonAPI編譯時先用文字編輯器開啟makefile,把python改成python3

然後執行如下指令(許可權不夠的話加上sudo)

make
make install
python setup.py install

3、生成model時提示:TypeError: softmax() got an unexpected keyword argument 'axis'

查詢後發現是Keras新版和repo中的語法衝突了,新版Keras去掉了這一引數。因此回退Keras版本到2.1。新版Keras附帶了preprocessing和applications兩個工具,為了解除安裝乾淨我先單獨卸掉這兩個工具,在執行如下命令重灌舊版本。

pip install keras==2.1

之後demo即可順利執行。