小麥穗數(shù)與小麥總產(chǎn)量息息相關(guān)，是重要的農(nóng)藝性狀之一。隨著人工智能、計算機視覺等信息技術(shù)的快速發(fā)展，已經(jīng)提出了不同的深度學(xué)習(xí)方法來分割和計數(shù)小麥穗。然而，當(dāng)前研究多集中在實驗室條件下，難以代表真實的、復(fù)雜的田間環(huán)境。由于麥田環(huán)境的多樣性，快速、準確地識別小麥穗數(shù)仍然是挑戰(zhàn)。
本研究旨在從小麥田間圖像中精準檢測、分割和計數(shù)小麥穗。首先整理并標注了一個具有精確掩碼和邊界框的小麥穗分割數(shù)據(jù)集SPIKE-segm（3種氮處理、不同生長階段下10個小麥品種圖像數(shù)據(jù)集），提出了一種基于Cascade Mask R-CNN架構(gòu)的、改進的穗分割模型，該模型對模型架構(gòu)中幾個組件進行了改進，包括可變形卷積網(wǎng)絡(luò)( DCN )，通用RoI提取器( GRoIE )、側(cè)邊感知邊界定位( SABL )和自適應(yīng)微調(diào)學(xué)習(xí)率LR。
為了評估這些組件的單獨和組合對模型性能的影響，設(shè)置了一個基于ResNet50網(wǎng)絡(luò)骨干的基線架構(gòu)，圖像尺寸為1333 × 800，batch size為2，LR為0.00125 (除非另有規(guī)定)。結(jié)果顯示，對于測試集，包含所有不同組件的版本在IoU = 0.5時獲得了最佳AP（平均精度），Bbox和Mask的AP分別為0.9303和0.9416。設(shè)定IoU閾值為0.5，統(tǒng)計每張圖像的TP，F(xiàn)P和FN，結(jié)果表明，該模型在測試圖像中平均準確率達到了86 %，平均F1分數(shù)為0.93。此外，與其他現(xiàn)有的架構(gòu)和策略相比，所提出的模型在穗檢測中mAP提高了0.41%，在穗分割任務(wù)中mAP提高了3.46%。由此可見，本研究所提出的小麥穗模型WheatSpikeNet為從田間圖像中準確分割小麥穗提供了高精度、高效、穩(wěn)定的方法，對未來小麥測產(chǎn)工作具有重要意義。
圖2 研究路線圖。描述了所提出的算法在田間圖像中檢測和分割小麥穗的工作流程。
圖4 SPIKE數(shù)據(jù)集的圖像標注示例( GSGC階段)。三種顏色類別：GSYC -綠穗，黃冠；GSGC -綠穗，綠冠；YSYC -黃穗，黃冠。
圖6本研究提出的穗分割模型結(jié)構(gòu)。
圖8 可變形卷積網(wǎng)絡(luò)。
圖9 側(cè)邊感知邊界定位( SABL )示例。
圖10 總體性能：( A )決定系數(shù)，真實值與檢測計數(shù)值的比較；( B )訓(xùn)練過程中的損失度量，超過150 epochs的訓(xùn)練損失；( C ) 超過150 epochs的訓(xùn)練精度；( D ) 超過150 epochs的驗證集mAP。
圖11 不同生長階段測試圖像真實值與檢測結(jié)果的可視化示例。( A ) GSGC真值圖；( B ) GSGC檢測結(jié)果；( C ) GSYC真值圖；( D ) GSYC檢測結(jié)果；( E ) YSYC真值圖；( F ) YSYC檢測結(jié)果。
表6 與現(xiàn)有研究方法的結(jié)果比較
文獻來源：Batin MA, Islam M, Hasan MM, Azad A, Alyami SA, Hossain MA and Miklavcic SJ (2023) WheatSpikeNet: an improved wheat spike segmentation model for accurate estimation from field imaging. Front. Plant Sci. 14:1226190. doi: 10.3389/fpls.2023.1226190
鄂公網(wǎng)安備42018502003245號 網(wǎng)站地圖
