在實(shí)際生產(chǎn)項(xiàng)目中，為了提升目標(biāo)識別的準(zhǔn)確性，我們往往需要3000-5000張圖片進(jìn)行標(biāo)注。而直接參與過標(biāo)注的人都有一個共同的感覺，就是標(biāo)注是一個簡單、枯燥、無聊且十分耗時費(fèi)力的差事。為此，我們可以在有了初步訓(xùn)練模型的基礎(chǔ)上，采用更加自動化的方式進(jìn)行標(biāo)注，屆時，你講不用每個照片、每個目標(biāo)的重復(fù)標(biāo)注，而是直接在現(xiàn)有模型對圖片進(jìn)行識別后的結(jié)果上進(jìn)行調(diào)整標(biāo)注或者新增標(biāo)注。本章節(jié)，筆者將重點(diǎn)圍繞使用labelIMG進(jìn)行自動化高效率的標(biāo)注進(jìn)行講解。

1、前期模型準(zhǔn)備

前期，筆者使用約1000張圖片對摩托車、電動自行車頭盔配搭和加裝雨棚進(jìn)行了訓(xùn)練，得到了一個已經(jīng)初步訓(xùn)練好的模型。暫且將其命名為“best_jx.pt”。下面我們將依托該pt文件模型，對新增的圖片進(jìn)行識別，并導(dǎo)入labelIMG進(jìn)行自動化標(biāo)準(zhǔn)，從而非常明顯的提升我們的圖片標(biāo)注工作效率。

相信，當(dāng)你讀到本文時，你也已經(jīng)有了自己的best.pt訓(xùn)練模型文件，大家可以直接套用本文檔的步驟。

2、編寫圖片識別腳本程序

當(dāng)你的圖片文件夾中已經(jīng)有了上千張最新的圖片后，我們現(xiàn)在要做的就是編寫代碼，調(diào)用前期你已經(jīng)生成的pt文件，將文件夾中的圖片逐一進(jìn)行識別。唯一不同的是，本次我們不光是要識別，還需要將識別的結(jié)果以標(biāo)簽的形式存為txt文檔。這個功能YOLO V8已經(jīng)在程序中實(shí)現(xiàn)了，我們只需要直接調(diào)用。

2.1 圖片標(biāo)簽生成程序

具體代碼示例如下：

from ultralytics