基于改進(jìn)生成對(duì)抗網(wǎng)絡(luò)的甜櫻桃數(shù)據(jù)增強(qiáng)方法
農(nóng)業(yè)機(jī)械學(xué)報(bào)
頁數(shù): 11 2024-07-31
摘要: 為解決在數(shù)據(jù)不平衡條件下甜櫻桃分類模型出現(xiàn)的長尾類不平衡問題,提出了一種基于深度卷積生成對(duì)抗網(wǎng)絡(luò)(Deep convolutional generative adversarial networks, DCGAN)的缺陷甜櫻桃圖像增強(qiáng)方法。首先,在生成器部分引入多尺度殘差塊(MSRB)和CBAM注意力機(jī)制,增強(qiáng)了模型特征表達(dá)能力和生成圖像細(xì)節(jié)質(zhì)量,同時(shí)改善了梯度流;在判別器部分... (共11頁)