融合注意力機(jī)制的解耦對比聚類
摘要: 為解決對比聚類正負(fù)樣本之間負(fù)正耦合的問題,提出融合注意力機(jī)制的解耦對比聚類DCCIAM方法。首先,使用數(shù)據(jù)增強(qiáng)手段將圖像數(shù)據(jù)進(jìn)行擴(kuò)充得到正樣本對和負(fù)樣本對;其次,在骨干網(wǎng)絡(luò)中加入卷積注意力模塊CBAM使網(wǎng)絡(luò)更加關(guān)注目標(biāo)特征,并將擴(kuò)充后的圖像數(shù)據(jù)輸入骨干網(wǎng)絡(luò)得到特征;再次,將特征經(jīng)過神經(jīng)網(wǎng)絡(luò)投影頭,分別計算實例損失和聚類損失;最后,聯(lián)合實例損失和聚類損失進(jìn)行特征表示及聚類分配。為... (共10頁)
開通會員,享受整站包年服務(wù)