之前環境配得一團糟,如何跑通maskRCNN的demo?

這兩天有個項目要用到mask_RCNN,之前用win10配了一下午,愣是跑不了demo,主要是因爲寫pycocotools的作者壓根就沒考慮windows用戶,真是巨坑。

下面提供一種非常可靠的方案,不管你之前環境配成什麼沙雕樣子,現在你按我的來,不出意外至少可以跑通demo。

這個方案第一點就是放棄windows,用linux來跑mask_RCNN。

樓主我正好有linux課程要做實驗,就乾脆裝了個Ubuntu第二系統在電腦上,感覺真是爽歪歪阿,但沒有想要長期用linux想法的,可以用windows自帶的ubuntu子系統來寫,或者用vmware虛擬機創建一個linux。

但windows自帶的子系統和虛擬機都有個致命缺陷,就是不能用gpu,只能用cpu,所以處理速度大概會比完整的系統慢幾十倍。

但僅入門而言,demo跑一張很複雜的對象分割圖也就第一次需要時間久一點,大概十多秒,後面就兩三秒。

配置mask_RCNN傳送門(保證三點:1.python用3.6版本,2.anaconda用anaconda3-4.1.3 3.tenserflow用cpu版本,也就是按照傳送門的做):https://www.cnblogs.com/herd/p/9364911.html

這麼弄完會出現一個問題,tenserflow1.5與keras2.1.6版本以上不兼容,到最後會出現一個錯誤:import name 'tf_utils' error

所以在運行jupyter notebook之前加一句:

pip install --upgrade keras==2.1.6

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章