mask-RCNN筆記——inspect_data使用
阿新 • • 發佈:2018-11-03
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