Tensorflow + Keras 執行Mask-RCNN
阿新 • • 發佈:2018-12-10
實驗環境: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即可順利執行。