1. 程式人生 > >影象語義分割標註工具labelme製作自己的資料集用於mask-rcnn訓練

影象語義分割標註工具labelme製作自己的資料集用於mask-rcnn訓練

labelme(標註mask資料集用的)

windows

python2

pip install pyqt
pip install labelme

python3

pip install pyqt5
pip install labelme

ubuntu16.04 系統自帶的python2.7環境

sudo apt-get install python-qt4 pyqt4-dev-tools
sudo pip install labelme  # python2 works

使用

(1)在終端中執行以下命令:

labelme

(2)出現labelme的介面:
這裡寫圖片描述

(3) 點選“open”,開啟需要標註的影象,選擇對目標區域進行標註

假如你要標註的物件為人和狗,在畫掩碼過程中,一幅影象中如果有多個person、dog,命名規則為person1、person2…… dog1、dog2……。因為labelme生成的標籤為一個label.png檔案,這個檔案只有一通道,在你標註時同一標籤mask會被給予一個標籤位,而mask要求不同的例項要放在不同的層中。最終訓練索要得到的輸入為一個w*h*n的ndarray,其中n為該圖片中例項的個數

這裡寫圖片描述

(4) 標註完成後點選,產生一個json檔案
這裡寫圖片描述

(5)進入<檔名>.json所在目錄下,在終端中執行

 labelme_json_to_dataset  <檔名>.json

這裡寫圖片描述

可得到一個資料夾,裡面有五個檔案,分別是:
*.png
info.yaml
label.png
label_names.txt
label_viz.png

這裡寫圖片描述

其中 label.png 和 info.yaml 是我們需要用到的! 標註已經完成!

label.png相當於mask 檔案,視覺化為:

這裡寫圖片描述

其中每個物件上的紅色數字,是我截圖的時候加上去的的,表示每個物件區域在影象中的真實畫素值

(6)如果需要批量轉換的話,就需要自己寫shell 指令碼

參考: