基于改進(jìn)FairMOT模型的多類別蠶繭跟蹤算法

摘要： 為獲取蠶繭智能分選過(guò)程中動(dòng)態(tài)群體蠶繭連續(xù)時(shí)間的種類及其坐標(biāo)位置信息，文中提出了一種基于改進(jìn)FairMOT模型的多類別蠶繭跟蹤算法。首先，按類別數(shù)擴(kuò)展預(yù)測(cè)頭的輸出維度，并構(gòu)建多類別損失函數(shù)，分別實(shí)現(xiàn)對(duì)上車?yán)O、黃斑繭和爛繭的跟蹤；其次，將原始FairMOT模型中的特征提取網(wǎng)絡(luò)更改為HarDNet網(wǎng)絡(luò)，并引入注意力機(jī)制，優(yōu)化蠶繭融合特征；最后，將特征權(quán)重同時(shí)輸入檢測(cè)和重識(shí)別分支進(jìn)行聯(lián)...