在colaboratory上配置
colaboratory上的包都是最新的,所以部分包需要降級纔可以使用
最關鍵的兩個是:
tensorflow1.3.0
keras2.0.8
這兩個包是我安裝的版本,版本不一過新,不然滿滿的bug折騰人。
-
安裝mask_rcnn:
我首先將mask rcnn給git到自己電腦的指定目錄下,然後上傳值colaboratory上
git clone https://github.com/matterport/Mask_RCNN
-
下載預訓練權重
mask_rcnn_coco.h5
下載好了之後需要將權重文件放到和demo.ipynb所在的同一個文件夾下
- 打開demo文件,裝載雲端硬盤
文件的目錄是:Mask_RCNN\samples\demo.ipynb
打開之後首先裝載googel雲端硬盤:
點擊這個之後會自動出現兩行代碼:
運行它,然後點擊這個給定的url:
進去之後會需要登錄,然後給定驗證碼,複製驗證碼,然後將驗證碼輸入在下面的輸入框中回車即可。
- 更換兩個包的版本:
看見上面的紅色字體,提示需要重新啓動,就點紅色字體下面的:
RESTART RUNTIME
等重新啓動之後,測試一下版本:
這裏提示一下,要將之前的tf-gpu刪除掉,不能cpu和gpu版本並存,因爲我試過。。。那錯誤賊離譜。。
版本對了,接着下面的操作
- 添加路徑
在文件裏需要導入若干包,但是我們還要添加包的路徑才能導入,爲了導入mrcnn文件夾裏的文件,需要將mrcnn所在的文件路徑加入:
import os
os.chdir('/content/drive/My Drive/Mask_RCNN')
!ls
注意不要一下子將那幾個包的路徑全部加進去,因爲路徑是會互相覆蓋的。
爲了導入model和coco我也做了如下的修改:
這些弄完了,需要提前注意圖片所在的路徑,同樣是在這個代碼單元裏,最後一行:
我創建一個frames1文件夾,裏面是待測試的圖片。
然後一個一個運行單元即可。