水稻病害診斷這件事聽起來(lái)好像就是個(gè)“拍照識(shí)圖”的小Demo但真正在田埂邊和實(shí)驗(yàn)室之間來(lái)回折騰過(guò)的人會(huì)明白它遠(yuǎn)比想象中要復(fù)雜。葉片病斑在自然光照下變化極大——同一種稻瘟病清晨帶露水、正午強(qiáng)光、陰雨連綿時(shí)拍出來(lái)顏色、紋理、邊緣銳利度完全是三個(gè)樣子。再加上YOLOv8這套框架背后的訓(xùn)練鏈路、數(shù)據(jù)清洗、推理部署、界面封裝每一步都是獨(dú)立的坑。這篇博客不打算復(fù)述“下載代碼跑起來(lái)”那種教程而是把整個(gè)項(xiàng)目從數(shù)據(jù)集倒騰到模型落地這一路的細(xì)節(jié)、參數(shù)依據(jù)和踩坑記錄都攤開來(lái)說(shuō)。無(wú)論你是想給畢設(shè)做一個(gè)可演示的系統(tǒng)還是想嘗試把檢測(cè)模型部署到農(nóng)業(yè)現(xiàn)場(chǎng)的嵌入式設(shè)備上這里的經(jīng)驗(yàn)和教訓(xùn)基本都能直接用上。項(xiàng)目本身由四部分構(gòu)成YOLOv8模型的訓(xùn)練源碼與推理源碼、一套經(jīng)過(guò)清洗和增強(qiáng)的水稻葉片病害圖像數(shù)據(jù)集以及一個(gè)基于桌面框架的可視化交互界面。它解決的核心問題是讓非計(jì)算機(jī)專業(yè)的使用者也能借助預(yù)訓(xùn)練模型對(duì)常見水稻葉片病害進(jìn)行快速篩查并對(duì)檢測(cè)結(jié)果進(jìn)行可視化存檔。1. 為什么選擇YOLOv8任務(wù)需求與技術(shù)選型的對(duì)照1.1 病害檢測(cè)對(duì)模型的核心要求在項(xiàng)目動(dòng)工之前先得想清楚檢測(cè)任務(wù)到底要什么。水稻葉片病害檢測(cè)面對(duì)的是復(fù)雜農(nóng)田環(huán)境植株密集、葉片交錯(cuò)、病斑大小不一同時(shí)存在大量相似的背景干擾泥土、水滴、其他雜草葉片。這意味著模型需要具備三方面素質(zhì)對(duì)微小病斑的敏感度、對(duì)自然光照變化的魯棒性、以及實(shí)時(shí)推理的能力。另外還有一個(gè)在實(shí)際項(xiàng)目中常被忽視卻極其重要的點(diǎn)——部署難度。如果做一個(gè)只能在實(shí)驗(yàn)室工作站上跑的模型它離“智能診斷系統(tǒng)”這個(gè)目標(biāo)還很遙遠(yuǎn)。真實(shí)的田間使用場(chǎng)景往往只有一臺(tái)普通性能的電腦甚至邊緣設(shè)備所以模型導(dǎo)出、量化、跨硬件運(yùn)行的能力應(yīng)當(dāng)在選型階段就納入考量而不是等模型訓(xùn)好了才發(fā)現(xiàn)部署不動(dòng)。這一點(diǎn)也是我最終確定圍繞YOLOv8來(lái)展開整個(gè)項(xiàng)目的重要原因之一。它的生態(tài)鏈覆蓋了從訓(xùn)練到導(dǎo)出、再到各種硬件平臺(tái)部署的完整路徑幾乎不存在“換環(huán)境重造輪子”的窘境。1.2 相比其他目標(biāo)檢測(cè)框架的取舍在選擇檢測(cè)框架時(shí)我對(duì)比了三類方案以Faster R-CNN為代表的兩階段檢測(cè)器、以SSD為代表的一階段早期模型、以及YOLO系列的近幾代版本。兩階段檢測(cè)器在精度上限上確實(shí)可觀尤其是對(duì)細(xì)節(jié)豐富的葉片病斑但其推理速度和工程復(fù)雜度在農(nóng)田現(xiàn)場(chǎng)多設(shè)備并發(fā)場(chǎng)景下不夠友好SSD的精度和召回表現(xiàn)又略顯不足特別是在小目標(biāo)方面。把范圍收窄到Y(jié)OLO系內(nèi)后YOLOv8對(duì)比上一代最明顯的幾個(gè)變化值得說(shuō)清楚。第一是特征提取網(wǎng)絡(luò)中的C2f模塊替代了原來(lái)的C3結(jié)構(gòu)它在保留輕量性的同時(shí)加強(qiáng)了梯度流動(dòng)梯度信息在深層網(wǎng)絡(luò)中不容易“斷流”對(duì)細(xì)小病斑的特征傳遞有實(shí)際幫助。第二是檢測(cè)頭從Anchor-Based改為Anchor-Free直接預(yù)測(cè)目標(biāo)中心距離省去了大量錨框超參數(shù)調(diào)試訓(xùn)練收斂更穩(wěn)定。第三是解耦檢測(cè)頭的使用分類和回歸分支獨(dú)立避免了兩者在反向傳播過(guò)程中的相互干擾。這些差異理論上聽起來(lái)很專業(yè)落地后最直接的感覺就是同批數(shù)據(jù)下mAP提升而且訓(xùn)練過(guò)程中l(wèi)oss曲線的震蕩明顯減少。1.3 生態(tài)與資料完備度的現(xiàn)實(shí)因素一個(gè)非常現(xiàn)實(shí)的因素是社區(qū)成熟度。YOLOv8的官方框架持續(xù)更新預(yù)訓(xùn)練權(quán)重豐富公開的教程和踩坑記錄很多這對(duì)項(xiàng)目推進(jìn)極為重要。我在訓(xùn)練過(guò)程中遇到過(guò)一個(gè)很奇怪的現(xiàn)象——大量回歸出的目標(biāo)框中心點(diǎn)偏移后來(lái)排查許久才發(fā)現(xiàn)是數(shù)據(jù)集中某一部分圖像Exif旋轉(zhuǎn)信息沒有歸一化而這個(gè)問題的解法在社區(qū)討論中早就有人總結(jié)過(guò)。如果換一個(gè)小眾框架可能連問題現(xiàn)象都描述不清楚。技術(shù)選型沒有絕對(duì)最優(yōu)只有適合自己和當(dāng)前任務(wù)的那一個(gè)。這個(gè)項(xiàng)目的核心是農(nóng)業(yè)場(chǎng)景下的病害檢測(cè)強(qiáng)調(diào)的是“能下場(chǎng)使用”的可靠性而不是單純爭(zhēng)奪學(xué)術(shù)榜單上的精確率小數(shù)點(diǎn)。YOLOv8在精度、速度、部署便利三方面的平衡性讓它成為這類項(xiàng)目最務(wù)實(shí)的選擇。2. 數(shù)據(jù)集構(gòu)建與預(yù)處理決定天花板的工程細(xì)節(jié)2.1 數(shù)據(jù)來(lái)源與版權(quán)邊界很多人拿到一個(gè)項(xiàng)目第一反應(yīng)是跑到公開數(shù)據(jù)集網(wǎng)站下載數(shù)據(jù)然后直接開始訓(xùn)練。這種流程在農(nóng)業(yè)病害檢測(cè)這類垂直場(chǎng)景里往往行不通——公開數(shù)據(jù)集的數(shù)量、類別覆蓋、拍攝環(huán)境與你的實(shí)際使用場(chǎng)景差別很大。我這次采用的方式是“公開數(shù)據(jù)集自采圖像”合并公開數(shù)據(jù)提供病害種類的基準(zhǔn)樣本自采圖像負(fù)責(zé)補(bǔ)充真實(shí)農(nóng)田中的光照、角度和背景多樣性。這里有個(gè)特別要提的注意事項(xiàng)數(shù)據(jù)集版權(quán)。不少公開數(shù)據(jù)集的授權(quán)協(xié)議只允許學(xué)術(shù)研究使用如果要構(gòu)建可分發(fā)給他人的完整系統(tǒng)比如作為畢設(shè)成果打包必須仔細(xì)閱讀授權(quán)條款。我在項(xiàng)目說(shuō)明中明確標(biāo)注了每個(gè)子集的來(lái)源和授權(quán)狀態(tài)避免后續(xù)傳播時(shí)引發(fā)版權(quán)問題。這一點(diǎn)建議你也盡早想清楚省得后期返工。2.2 類別定義與標(biāo)注細(xì)節(jié)水稻葉片病害的種類依據(jù)項(xiàng)目需求圈定為四類稻瘟病葉片上的梭形病斑、紋枯病云紋狀病斑、白葉枯病葉緣黃化枯死、胡麻斑病芝麻粒大小的褐色斑點(diǎn)另外單獨(dú)設(shè)置一個(gè)“健康葉片”類別用于對(duì)比判斷。類別數(shù)量不宜過(guò)多否則樣本量會(huì)被稀釋模型在小樣本類別上更容易過(guò)擬合。標(biāo)注是整個(gè)過(guò)程中最枯燥但最影響結(jié)果上限的環(huán)節(jié)。我使用LabelImg工具逐張框選病斑區(qū)域遵循幾個(gè)硬性規(guī)則病斑與健康組織邊界不清晰時(shí)標(biāo)注框包含完整的病斑區(qū)域以及少量健康過(guò)渡帶單張葉片的多個(gè)分散病斑分別標(biāo)注不合并為大框遮擋嚴(yán)重的病斑不標(biāo)注避免引入標(biāo)注噪聲每張圖像標(biāo)注完畢后隨機(jī)抽樣復(fù)查與第二人交叉檢查標(biāo)注一致性。驗(yàn)證集和測(cè)試集的劃分發(fā)生在標(biāo)注完成之后且要保證來(lái)自同一株水稻的不同圖像不會(huì)同時(shí)出現(xiàn)在訓(xùn)練集和驗(yàn)證集中——否則模型會(huì)在驗(yàn)證集上“作弊”表現(xiàn)為驗(yàn)證集指標(biāo)虛高而真實(shí)場(chǎng)景效果落差很大。2.3 數(shù)據(jù)增強(qiáng)策略的實(shí)驗(yàn)選擇YOLOv8在訓(xùn)練時(shí)會(huì)自動(dòng)啟用Mosaic增強(qiáng)把四張圖隨機(jī)拼接到一張上對(duì)小目標(biāo)的檢測(cè)能力提升明顯也讓模型對(duì)目標(biāo)被截?cái)嗟那闆r更有容忍度。不過(guò)Mosaic增強(qiáng)在訓(xùn)練后期存在一個(gè)隱患拼接圖上的目標(biāo)比例與真實(shí)圖像分布不符持續(xù)使用可能導(dǎo)致模型在真實(shí)圖像上精度下降。實(shí)際的做法是前一半訓(xùn)練輪次開啟Mosaic后一半訓(xùn)練中將Mosaic概率降低甚至關(guān)閉讓模型在接近真實(shí)分布的數(shù)據(jù)上微調(diào)收斂。除了內(nèi)置增強(qiáng)我還額外加入了亮度擾動(dòng)、色度擾動(dòng)和隨機(jī)仿射變換。做這個(gè)選擇的原因很直接——水稻病害的視覺特征受環(huán)境光照影響極大模型如果只看過(guò)光線均勻的圖片放在室外自然光下一測(cè)精度會(huì)掉得讓人崩潰。增強(qiáng)參數(shù)的設(shè)置不能過(guò)于激進(jìn)比如亮度擾動(dòng)幅度控制在正負(fù)20%區(qū)間內(nèi)否則病斑的顏色特征會(huì)被破壞反而讓模型學(xué)不到關(guān)鍵差異。2.4 數(shù)據(jù)目錄結(jié)構(gòu)與訓(xùn)練集劃分整個(gè)數(shù)據(jù)目錄按YOLOv8的習(xí)慣整理成一個(gè)標(biāo)準(zhǔn)結(jié)構(gòu)后續(xù)訓(xùn)練和推理都能直接讀取。datasets/ ├── rice_disease/ │ ├── images/ │ │ ├── train/ │ │ ├── val/ │ │ └── test/ │ ├── labels/ │ │ ├── train/ │ │ ├── val/ │ │ └── test/ │ └── data.yaml最終數(shù)據(jù)集包含大約八千張圖像其中訓(xùn)練集占七成驗(yàn)證集與測(cè)試集各占一成半。各個(gè)病害類別之間的樣本數(shù)量盡量對(duì)齊保證模型不會(huì)因?yàn)槠么箢惗鲆曅☆?。如果你打算沿用這套結(jié)構(gòu)最需要注意的一點(diǎn)是標(biāo)簽文件中的類別編號(hào)必須與data.yaml中的類別列表嚴(yán)格一致否則訓(xùn)練時(shí)會(huì)遇到類別錯(cuò)亂的報(bào)錯(cuò)排查起來(lái)很費(fèi)神。3. YOLOv8訓(xùn)練避坑從環(huán)境配置到收斂評(píng)估的全程記錄3.1 環(huán)境與版本選擇YOLOv8的訓(xùn)練代碼基于Ultralytics框架安裝依賴時(shí)有一個(gè)經(jīng)常被忽視的細(xì)節(jié)——版本鎖定。我使用的是官方文檔中推薦的一組固定版本組合而不是盲目安裝最新版。深度學(xué)習(xí)框架之間的依賴關(guān)系向來(lái)微妙PyTorch、CUDA、顯卡驅(qū)動(dòng)與Ultralytics的版本矩陣如果不匹配輕則警告不斷重則直接無(wú)法使用GPU訓(xùn)練。如果你手頭的顯卡顯存小于8G建議直接使用YOLOv8n或YOLOv8s這樣的輕量級(jí)模型作為起點(diǎn)而不是一上來(lái)就嘗試最大的版本。大模型對(duì)小批量下的顯存需求很敏感一旦顯存溢出不一定報(bào)錯(cuò)而是表現(xiàn)為訓(xùn)練速度驟降和系統(tǒng)無(wú)響應(yīng)。在這種場(chǎng)景下最穩(wěn)妥的方案是減少批次大小同時(shí)保持輸入尺寸不變而不是把輸入尺寸縮小——后者會(huì)犧牲對(duì)小病斑的檢測(cè)能力。3.2 訓(xùn)練參數(shù)的設(shè)定依據(jù)訓(xùn)練要設(shè)定的核心參數(shù)包括輸入尺寸、每批樣本數(shù)、訓(xùn)練輪次、初始學(xué)習(xí)率和權(quán)重衰減系數(shù)。這次項(xiàng)目使用的是輸入尺寸640像素、批次大小16、訓(xùn)練輪次200、初始學(xué)習(xí)率0.01權(quán)重衰減系數(shù)0.0005的組合。輸入尺寸的選擇基于數(shù)據(jù)集圖像的實(shí)際分辨率——大多數(shù)手機(jī)的拍攝照片遠(yuǎn)大于640像素直接縮放會(huì)丟失病斑細(xì)節(jié)但如果統(tǒng)一使用1280像素輸入訓(xùn)練時(shí)間和顯存需求又成倍增加。折中方案是640像素輸入加上Mosaic增強(qiáng)讓模型在計(jì)算代價(jià)可控的前提下盡可能看到小目標(biāo)。3.3 訓(xùn)練過(guò)程的監(jiān)控指標(biāo)訓(xùn)練啟動(dòng)后不要只盯著mAP一個(gè)數(shù)字。我習(xí)慣每五個(gè)輪次記錄一組直觀指標(biāo)訓(xùn)練集loss、驗(yàn)證集loss、精確率、召回率和各時(shí)段mAP值。沒有收斂判斷經(jīng)驗(yàn)的同學(xué)最容易犯的錯(cuò)誤是認(rèn)為“l(fā)oss越低越好”。實(shí)際上訓(xùn)練集loss不斷下降而驗(yàn)證集loss上翹就是過(guò)擬合的典型信號(hào)。在實(shí)際訓(xùn)練中模型在初始幾個(gè)輪次中mAP上升很慢比如前30個(gè)輪次mAP50一直在0.3附近徘徊這時(shí)不要急著調(diào)參或中斷訓(xùn)練通常到第60個(gè)輪次后會(huì)有明顯跳變。這是YOLO系列模型訓(xùn)練的正常節(jié)奏——前期的特征提取層和檢測(cè)頭參數(shù)都在磨合效果會(huì)有一個(gè)從“幾乎不工作”到“突然可用”的飛躍。3.4 評(píng)估指標(biāo)的分析訓(xùn)練完成后用測(cè)試集檢查模型。測(cè)試集的意義不同于驗(yàn)證集——驗(yàn)證集參與了早停判斷和超參調(diào)優(yōu)測(cè)試集則是完全獨(dú)立的衡量尺度。測(cè)試集評(píng)估結(jié)果顯示四類病害的mAP50值在0.82到0.91之間白葉枯病效果最好胡麻斑病的檢測(cè)指標(biāo)相對(duì)偏低。原因也不難理解胡麻斑病的病斑尺寸太小在640像素的輸入下一個(gè)只有十幾個(gè)像素大小的斑點(diǎn)幾乎退化成了噪點(diǎn)。這時(shí)我配合工具分析模型在各尺寸目標(biāo)上的表現(xiàn)分布確認(rèn)小尺寸目標(biāo)占據(jù)了大多數(shù)誤檢。解決思路比較清晰——在使用模型做實(shí)際推理時(shí)增加了一個(gè)切片推理步驟將大圖切分為若干帶重疊區(qū)域的子圖分別在子圖上檢測(cè)再把結(jié)果合并回原圖坐標(biāo)。這一步的代價(jià)是推理耗時(shí)上升但換來(lái)了小病斑檢測(cè)率的顯著提升對(duì)水稻葉片這類密度高、病斑小的場(chǎng)景來(lái)說(shuō)非常值得。4. 推理鏈路與可視化界面開發(fā)從權(quán)重文件到可交互系統(tǒng)4.1 整體架構(gòu)設(shè)計(jì)訓(xùn)練產(chǎn)出的best.pt權(quán)重文件只是一個(gè)裸模型離“智能診斷系統(tǒng)”這個(gè)詞還差一個(gè)完整的應(yīng)用層。我采用的架構(gòu)是將推理封裝成一個(gè)獨(dú)立的檢測(cè)服務(wù)模塊與界面層通過(guò)回調(diào)函數(shù)通信。這樣設(shè)計(jì)的好處是未來(lái)如果想把這個(gè)檢測(cè)模塊移植到命令行工具或嵌入式設(shè)備上不用改動(dòng)界面代碼復(fù)用性更強(qiáng)。檢測(cè)模塊需要完成的核心流程是讀取輸入圖片、根據(jù)圖像尺寸決定是否切片推理、對(duì)每個(gè)切片執(zhí)行推理、將切片級(jí)別的檢測(cè)框映射回原圖坐標(biāo)、應(yīng)用NMS去重、附加類別標(biāo)簽和置信度分?jǐn)?shù)。切片推理的結(jié)果合并看似簡(jiǎn)單實(shí)際操作中有不少細(xì)節(jié)。切片之間的重疊區(qū)域會(huì)產(chǎn)生重復(fù)檢測(cè)框NMS的閾值設(shè)置、重疊區(qū)域的寬度設(shè)置都會(huì)影響最終結(jié)果質(zhì)量。經(jīng)過(guò)多次試驗(yàn)我將切片尺寸設(shè)為原圖的一半重疊寬度設(shè)置為切片寬度的10%NMS閾值采用0.45整套流程在小病斑場(chǎng)景下效果表現(xiàn)很穩(wěn)。4.2 模型導(dǎo)出與部署優(yōu)化策略桌面端推理模塊的模型格式選擇直接影響了部署環(huán)境的復(fù)雜度。PyTorch格式雖然便于二次開發(fā)但需要額外安裝對(duì)應(yīng)版本的PyTorch和Ultralytics框架對(duì)普通用戶不友好。我最終導(dǎo)出了ONNX格式的模型文件配合開放神經(jīng)網(wǎng)絡(luò)交換格式的推理引擎做運(yùn)行時(shí)顯著減少了運(yùn)行環(huán)境的依賴體積。ONNX導(dǎo)出的過(guò)程中一個(gè)常用技巧是固定模型的動(dòng)態(tài)軸參數(shù)。YOLOv8默認(rèn)支持動(dòng)態(tài)輸入尺寸但這會(huì)導(dǎo)致某些推理引擎無(wú)法預(yù)先分配內(nèi)存增加延遲。我在導(dǎo)出時(shí)手動(dòng)指定輸入尺寸為1×3×640×640換取更快的推理速度。導(dǎo)出的ONNX模型在CPU上處理一張普通分辨率的圖像耗時(shí)大約在300到500毫秒之間實(shí)際體驗(yàn)是“點(diǎn)擊檢測(cè)后能看到明顯的加載狀態(tài)但不算卡頓”。4.3 可視化界面的實(shí)現(xiàn)方案界面層選用PySide6Qt的Python綁定來(lái)實(shí)現(xiàn)主要考慮三點(diǎn)跨平臺(tái)、控件庫(kù)成熟、支持圖像實(shí)時(shí)顯示。系統(tǒng)界面布局分為三個(gè)功能區(qū)域左側(cè)為圖片選擇與參數(shù)控制面板中間為檢測(cè)結(jié)果預(yù)覽區(qū)域右側(cè)為檢測(cè)統(tǒng)計(jì)信息面板還額外加了一個(gè)識(shí)別記錄導(dǎo)出按鈕。用戶可以通過(guò)界面選擇單張圖片、批量處理一個(gè)文件夾的圖像也支持接入攝像頭做實(shí)時(shí)檢測(cè)。界面與檢測(cè)模塊之間的數(shù)據(jù)交換用的是信號(hào)槽機(jī)制檢測(cè)過(guò)程放到子線程中執(zhí)行避免主界面在檢測(cè)耗時(shí)較長(zhǎng)時(shí)出現(xiàn)假死現(xiàn)象。界面下方的統(tǒng)計(jì)信息表格會(huì)列出每張圖片中各類病害的檢出數(shù)量一鍵導(dǎo)出為CSV文件。這個(gè)功能在實(shí)際使用中很受歡迎因?yàn)橹脖Ｈ藛T做田間調(diào)查時(shí)不僅想知道“有沒有病”更想記錄“有多少病”。界面核心代碼如下from PySide6.QtWidgets import (QMainWindow, QLabel, QPushButton, QFileDialog, QVBoxLayout, QHBoxLayout, QTableWidget, QWidget) from PySide6.QtCore import Qt, QThread, Signal class DetectThread(QThread): result_ready Signal(dict) def __init__(self, model_path, image_path): super().__init__() self.model_path model_path self.image_path image_path def run(self): # 調(diào)用ONNX推理模塊 result run_inference(self.model_path, self.image_path) self.result_ready.emit(result) class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle(水稻葉片病害智能診斷系統(tǒng)) self.model_path best.onnx self._init_ui() def _init_ui(self): # 構(gòu)建界面布局 central QWidget() layout QHBoxLayout(central) left_panel QVBoxLayout() self.select_btn QPushButton(選擇圖片) self.detect_btn QPushButton(開始檢測(cè)) self.table QTableWidget() left_panel.addWidget(self.select_btn) left_panel.addWidget(self.detect_btn) left_panel.addWidget(self.table) self.image_label QLabel(請(qǐng)?jiān)谧髠?cè)選擇圖片) self.image_label.setAlignment(Qt.AlignmentFlag.AlignCenter) layout.addLayout(left_panel, 1) layout.addWidget(self.image_label, 3) central.setLayout(layout) self.setCentralWidget(central)4.4 實(shí)際效果復(fù)盤整個(gè)系統(tǒng)完成之后我請(qǐng)幾位做植物保護(hù)研究的同行拿他們自己采集的田間照片做了一輪盲測(cè)。盲測(cè)對(duì)象包含約兩百?gòu)埐煌庹窄h(huán)境、不同手機(jī)拍攝參數(shù)的照片系統(tǒng)在健康葉片上的誤報(bào)率較低四類病害的總體檢出率為百分之八十六左右。最影響體驗(yàn)的場(chǎng)景是強(qiáng)逆光拍攝的葉片照片病斑區(qū)域嚴(yán)重曝光不足模型幾乎無(wú)法提取有效特征。界面端暫時(shí)沒有設(shè)計(jì)圖像增強(qiáng)功能我建議用戶在實(shí)際使用中盡量選擇順光拍攝。5. 高頻踩坑記錄給打算復(fù)現(xiàn)的同學(xué)省點(diǎn)時(shí)間5.1 小病斑漏檢與誤檢的平衡整個(gè)項(xiàng)目中最難纏的問題是小病斑檢測(cè)。胡麻斑病的斑點(diǎn)數(shù)量多、體積小幾個(gè)像素級(jí)別的斑點(diǎn)堆疊在一起如果Mosaic增強(qiáng)分配的小目標(biāo)訓(xùn)練樣本不足模型很容易將它們整體忽略。試過(guò)大幅調(diào)高輸入分辨率但訓(xùn)練速度因此慢了接近一倍代價(jià)過(guò)高。最終收斂的方案是兩層組合訓(xùn)練階段調(diào)節(jié)訓(xùn)練數(shù)據(jù)的拼圖增廣比例并引入了目標(biāo)尺寸自適應(yīng)的復(fù)制粘貼增強(qiáng)——把小病斑樣本額外粘貼到無(wú)病斑的葉片圖像上相當(dāng)于在訓(xùn)練層面人為增加小目標(biāo)的數(shù)量。配合推理階段的切片策略最終把小病斑類別的平均檢出率提高了將近十個(gè)百分點(diǎn)。5.2 數(shù)據(jù)噪聲帶來(lái)的不可復(fù)現(xiàn)問題另一個(gè)容易讓人崩潰的問題是同一個(gè)代碼、同一份數(shù)據(jù)兩次訓(xùn)練得到的結(jié)果波動(dòng)明顯。一開始我以為是隨機(jī)種子的問題但排查后發(fā)現(xiàn)問題出在數(shù)據(jù)增強(qiáng)的隨機(jī)性影響過(guò)大部分增強(qiáng)操作對(duì)小尺寸病斑產(chǎn)生了破壞性的形變導(dǎo)致模型在部分類別的學(xué)習(xí)極不穩(wěn)定。解決方案是削減增強(qiáng)強(qiáng)度上限同時(shí)固定全局隨機(jī)種子訓(xùn)練就變得可控了多次運(yùn)行的結(jié)果波動(dòng)幅度明顯減小。5.3 界面開發(fā)中被忽視的路徑問題如果你要復(fù)用這套界面源碼有三類路徑相關(guān)的問題大概率會(huì)遇到包含中文的路徑、網(wǎng)絡(luò)驅(qū)動(dòng)器路徑、以及圖像命名中的特殊字符。部分推理引擎對(duì)中文路徑支持不佳會(huì)導(dǎo)致模型文件加載失敗而這類錯(cuò)誤在終端里看不到明顯報(bào)錯(cuò)只表現(xiàn)為界面按鈕無(wú)響應(yīng)。解決方案非常樸素在項(xiàng)目目錄規(guī)范文檔中強(qiáng)制約定項(xiàng)目路徑不能包含中文與空格同時(shí)在代碼中統(tǒng)一使用原始字符串和路徑標(biāo)準(zhǔn)化處理。另一個(gè)隱蔽的問題是圖像文件擴(kuò)展名大小寫混用。JPG與jpeg在Windows和Linux下的識(shí)別行為不一致導(dǎo)致批量檢測(cè)時(shí)個(gè)別文件被跳過(guò)。處理方式是代碼中統(tǒng)一轉(zhuǎn)換為小寫判斷。5.4 硬件與運(yùn)行環(huán)境的適配如果你打算把系統(tǒng)交給非技術(shù)背景的使用者運(yùn)行環(huán)境搭建的成本會(huì)比你預(yù)想的高得多。電腦缺少顯卡驅(qū)動(dòng)或CUDA版本不合適系統(tǒng)會(huì)自動(dòng)回退到CPU推理但速度大幅下降。圖形界面運(yùn)行時(shí)CPU占用率甚至?xí)咏鼭M載。我的解決方法是在系統(tǒng)啟動(dòng)時(shí)自動(dòng)檢測(cè)推理后端并在界面狀態(tài)欄顯示當(dāng)前是GPU模式還是CPU模式同時(shí)提醒用戶安裝必要組件。這個(gè)小細(xì)節(jié)讓整個(gè)系統(tǒng)交付后的售后問題減少了很多值得推薦。6. 從畢業(yè)設(shè)計(jì)到產(chǎn)品原型的擴(kuò)展思路這個(gè)項(xiàng)目目前的狀態(tài)是一套可以直接運(yùn)行的原型系統(tǒng)但如果想沿這個(gè)方向繼續(xù)推進(jìn)有幾個(gè)明確的擴(kuò)展路徑。參考我給同行提建議的思路接下來(lái)值得試驗(yàn)的方向包括加入更細(xì)粒度的病害分級(jí)功能比如輕度、中度、重度感染因?yàn)橹脖Ｈ藛T在田間更關(guān)心的是“該用什么程度的藥劑干預(yù)”而不只是“得沒得病”將模型輕量化后部署到無(wú)人機(jī)或手機(jī)端巡檢效率可以提升一個(gè)量級(jí)。光照魯棒性、復(fù)雜背景干擾、跨品種適應(yīng)性都是真實(shí)田間環(huán)境對(duì)算法提出的新考題而這些問題需要在農(nóng)業(yè)場(chǎng)景第一線反復(fù)測(cè)試、持續(xù)迭代才能讓這套系統(tǒng)從“能跑”變成“可靠”。最后再分享一條親身經(jīng)驗(yàn)整個(gè)項(xiàng)目中真正幫你建立能力邊界的不是跑通某個(gè)現(xiàn)成模型而是動(dòng)手整理數(shù)據(jù)、反復(fù)查看檢測(cè)錯(cuò)誤樣例、理解模型在哪些情況下“束手無(wú)策”的過(guò)程。別急著追求指標(biāo)上的好看數(shù)字多去田間拍幾張不同條件下的照片交給系統(tǒng)試一試——那些失敗案例才是把系統(tǒng)打磨扎實(shí)的階梯。