前言:在使用YOLO訓(xùn)練自己模型的時(shí)候首先要學(xué)會(huì)對(duì)數(shù)據(jù)進(jìn)行處理,這里介紹一個(gè)常用的本地打標(biāo)簽工具labelimg,如果不想按照的話也可以使用在線標(biāo)簽工具,因?yàn)楹ε聰?shù)據(jù)泄露,所以本人一直使用的是本地工具進(jìn)行打標(biāo)簽。在線標(biāo)注工具網(wǎng)站:https://www.makesense.ai/
1、labelimg標(biāo)簽工具介紹
Labelimg是一款開源的數(shù)據(jù)標(biāo)注工具,可以標(biāo)注三種格式。
- VOC標(biāo)簽格式,保存為xml文件。
- yolo標(biāo)簽格式,保存為txt文件。
- createML標(biāo)簽格式,保存為json格式。
2、labelimg的安裝
這里主要講的是在window系統(tǒng)中的安裝,首先打開cmd命令行(快捷鍵:win+R)。進(jìn)入cmd命令行控制臺(tái)。輸入如下的命令:
pip install labelimg -i https://pypi.tuna.tsinghua.edu.cn/simple
運(yùn)行如上命令后,系統(tǒng)就會(huì)自動(dòng)下載labelimg相關(guān)的依賴。由于這是一個(gè)很輕量的工具,所以下載起來很快,當(dāng)出現(xiàn)如下紅色下劃線的時(shí)候,告訴我們成功安裝的時(shí)候,說明labelimg安裝成功了。
3、如何使用labelimg工具
3.1據(jù)準(zhǔn)備
首先這里需要準(zhǔn)備我們需要打標(biāo)注的數(shù)據(jù)集。這里我建議新建一個(gè)名為img_data的文件夾(這個(gè)是約定俗成,不這么做也行),里面創(chuàng)建一個(gè)名為images的文件夾存放我們需要打標(biāo)簽的圖片文件;再創(chuàng)建一個(gè)名為labels存放標(biāo)注的標(biāo)簽文件;最后創(chuàng)建一個(gè)名為 classes.txt 的txt文件來存放所要標(biāo)注的類別名稱。
img_data的目錄結(jié)構(gòu)如下:
|—img_data
||──images 存放需要打標(biāo)簽的圖片文件
||── labels 存放標(biāo)注的標(biāo)簽文件
||── classes.txt 定義自己要標(biāo)注的所有類別(這個(gè)文件可有可無,但是在我們定義類別比較多的時(shí)候,最好有這個(gè)創(chuàng)建一個(gè)這樣的txt文件來存放類別)
3.2標(biāo)注前設(shè)置
首先在images這個(gè)文件夾放置待標(biāo)注的圖片,這里是三類圖片,分別是人、狗和貓。然后再classes.txt 這個(gè)txt文檔里面輸入定義的類別種類;如下圖所示。
進(jìn)入img_data文件夾當(dāng)中,在搜索欄里面輸入cmd打開終端命令(這一步非常重要,如果不是在這個(gè)路徑下打開終端將不能利用txt文件中的類別);或者先打開cmd命令終端(快捷鍵:win+R),再使用cd命令進(jìn)入到txt文本這個(gè)文件夾當(dāng)中;
輸入如下的命令打開labelimg。這個(gè)命令的意思是打開labelimg工具;打開images文件夾,初始化classes.txt里面定義的類。
labelimg images classes.txt
運(yùn)行上面的命令就會(huì)打開這個(gè)labelimg工具如下:
3.2.1按鈕設(shè)置
下面介紹圖中的我們常用的按鈕
點(diǎn)擊View,會(huì)出現(xiàn)如下選項(xiàng),盡量和我一樣去勾選
- Auto Save mode:切換到下一張圖的時(shí)候,會(huì)自動(dòng)保存標(biāo)簽
- Display Labels:會(huì)顯示標(biāo)注框和標(biāo)簽
- Advanced Mode:標(biāo)注的十字架會(huì)一直懸浮在窗口
3.2.2常見快捷鍵
A:切換到上一張圖片
D:切換到下一張圖片
W:調(diào)出標(biāo)注十字架
del :刪除標(biāo)注框框
Ctrl+u:選擇標(biāo)注的圖片文件夾
Ctrl+r:選擇標(biāo)注好的label標(biāo)簽存在的文件夾文章來源:http://www.zghlxwxcb.cn/news/detail-522016.html
3.3開始標(biāo)注
由于我們?cè)O(shè)置標(biāo)注的十字架一直在標(biāo)注界面上,這就不需要我們按快捷鍵w,然后選定我們需要標(biāo)注的對(duì)象。按住鼠標(biāo)左鍵拖出框框就可以了。如下圖所示,當(dāng)我們選定目標(biāo)以后,就會(huì)加載出來classes.txt 定義自己要標(biāo)注的所有類別(如果類別多,是真的很方便,就不需要自己手打每個(gè)類別的名字了)。打好的標(biāo)簽框框上會(huì)有該框框的類別。然后界面最右邊會(huì)出現(xiàn)打好的類別標(biāo)簽。打好一張照片以后,快捷鍵D,就會(huì)進(jìn)入下一張,這時(shí)候就會(huì)自動(dòng)保存標(biāo)簽文件(voc格式會(huì)保存xml,yolo會(huì)保存txt格式)。
當(dāng)一張圖片上面又多個(gè)物體時(shí)可以打多個(gè)標(biāo)注:
打完標(biāo)簽之后進(jìn)入到labels文件夾當(dāng)中,可以發(fā)現(xiàn)生成了多個(gè)txt文件
點(diǎn)擊進(jìn)去之后會(huì)發(fā)現(xiàn)生成一些數(shù)據(jù)文件,第一個(gè)0代表我們打標(biāo)簽的種類,第0類為person類,第1為dog類,后面的4個(gè)數(shù)據(jù)記錄的是打標(biāo)注的位置。
下篇文章介紹如何進(jìn)行數(shù)據(jù)集格式轉(zhuǎn)換和劃分;文章來源地址http://www.zghlxwxcb.cn/news/detail-522016.html
到了這里,關(guān)于YOLOV5目標(biāo)檢測(cè)---labelimg圖片標(biāo)注工具(1)的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!