1. 程式人生 > >mask-RCNN筆記——inspect_data使用

mask-RCNN筆記——inspect_data使用

mask-RCNN的使用,其中程式碼inspect_data的使用,用於dateset的讀取

1、下載coco資料庫,安裝coco工具包pycocotools

https://blog.csdn.net/Diana_Z/article/details/83576598

2、新建一個*.py檔案將*。ipynb的程式碼複製過去

3、資料路徑修改:

修改程式碼中下圖紅框處COCO_DIR變數的值為資料資料夾dataset的路徑。

執行

1、資料讀取

2、隨機顯示四張圖片以及圖片類掩碼

3、隨機一張圖片顯示掩碼以及類選框,並輸出資訊

image_id:表示圖片在dataset的圖片列表中的下標

image和mask是兩張與原圖一樣大小的圖片,image是原圖,mask是疊加之後的掩碼圖片

class_ids:是一個3*1的dnarry,表示有3個選框分別對應三個類id

bbox:是一個3*4的dnarry,每行表示一個選框的位置

4、隨機一張圖片顯示resize後的圖片以及分類和掩碼,

圖片大小統一到1024

圖片經過縮放並填充

5、輸出預選礦的情況anchors

6、輸出一個cell的anchor情況

7、顯示候選框以及最終的結果

正候選框(positive anchors)

負候選框(negative anchors)

其他選框,沒有加入訓練中

8、ROI