摘要:水稻籽粒檢測在糧食儲存中凸顯重要作用,直接影響糧食銷售的價格,。針對一般機(jī)器視覺檢測算法在水稻籽粒小目標(biāo)的密集場景下存在難以識別且網(wǎng)絡(luò)模型參數(shù)大,,檢測速度較慢、成本高等問題,,提出一種基于YOLO v7優(yōu)化的水稻籽粒檢測算法,。首先將部分高效聚合網(wǎng)絡(luò)模塊(Efficient layer aggregation network,ELAN)替換成輕量級網(wǎng)絡(luò)模塊GhostNetV2添加到主干及頸部網(wǎng)絡(luò)部分,實現(xiàn)網(wǎng)絡(luò)參數(shù)精簡化的同時也減少了通道中的特征冗余,;其次將卷積和自注意力結(jié)合的注意力模塊(Convolution and self-attention mixed model,ACmix)添加到MP模塊中,,平衡全局和局部的特征信息,,充分關(guān)注特征映射的細(xì)節(jié)信息;最后使用WIoU(Wise intersection over union)作為損失函數(shù),,減少了距離,、縱橫比之類的懲罰項干擾,單調(diào)聚焦機(jī)制的設(shè)計提高了模型的定位性能,。在水稻籽粒圖像數(shù)據(jù)集上驗證改進(jìn)后的模型檢測水平,,實驗結(jié)果表明,改進(jìn)后的YOLO v7模型的[email protected]達(dá)96.55%,,[email protected]:0.95達(dá)70.10%,,訓(xùn)練模型參數(shù)量也有所下降,在實際場景以暗黑色為背景的水稻雜質(zhì)檢測中的效果優(yōu)于其他模型,,滿足了水稻籽粒的實時檢測要求,,可將此算法應(yīng)用于自動化檢測糧食系統(tǒng)中。