配置mask rcnn環境並運行源碼

在colaboratory上配置

colaboratory上的包都是最新的,所以部分包需要降級纔可以使用
最關鍵的兩個是:
tensorflow1.3.0
keras
2.0.8
這兩個包是我安裝的版本,版本不一過新,不然滿滿的bug折騰人。

  1. 安裝mask_rcnn:
    我首先將mask rcnn給git到自己電腦的指定目錄下,然後上傳值colaboratory上
    git clone https://github.com/matterport/Mask_RCNN

  2. 下載預訓練權重
    mask_rcnn_coco.h5

下載好了之後需要將權重文件放到和demo.ipynb所在的同一個文件夾下

  1. 打開demo文件,裝載雲端硬盤
    文件的目錄是:Mask_RCNN\samples\demo.ipynb
    打開之後首先裝載googel雲端硬盤:
    在這裏插入圖片描述

點擊這個之後會自動出現兩行代碼:
在這裏插入圖片描述

運行它,然後點擊這個給定的url:
在這裏插入圖片描述

進去之後會需要登錄,然後給定驗證碼,複製驗證碼,然後將驗證碼輸入在下面的輸入框中回車即可。

  1. 更換兩個包的版本:

在這裏插入圖片描述

看見上面的紅色字體,提示需要重新啓動,就點紅色字體下面的:
RESTART RUNTIME
等重新啓動之後,測試一下版本:
在這裏插入圖片描述

這裏提示一下,要將之前的tf-gpu刪除掉,不能cpu和gpu版本並存,因爲我試過。。。那錯誤賊離譜。。
版本對了,接着下面的操作

  1. 添加路徑

在這裏插入圖片描述

在文件裏需要導入若干包,但是我們還要添加包的路徑才能導入,爲了導入mrcnn文件夾裏的文件,需要將mrcnn所在的文件路徑加入:

import os
os.chdir('/content/drive/My Drive/Mask_RCNN')
!ls

在這裏插入圖片描述

注意不要一下子將那幾個包的路徑全部加進去,因爲路徑是會互相覆蓋的。

爲了導入model和coco我也做了如下的修改:
在這裏插入圖片描述

這些弄完了,需要提前注意圖片所在的路徑,同樣是在這個代碼單元裏,最後一行:
在這裏插入圖片描述

我創建一個frames1文件夾,裏面是待測試的圖片。

然後一個一個運行單元即可。

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