基于改進(jìn)FairMOT模型的多類別蠶繭跟蹤算法
蠶業(yè)科學(xué)
頁數(shù): 9 2023-08-15
摘要: 為獲取蠶繭智能分選過程中動(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)... (共9頁)