簡介這是一套面向玉米葉病害檢測的完整資源整合了YOLOv8訓(xùn)練權(quán)重、PyQt可視化界面以及一千五百張已標(biāo)注數(shù)據(jù)集。數(shù)據(jù)集預(yù)先劃分好train、val、test三個子集并附帶data.yaml配置標(biāo)簽采用txt格式可直接用于YOLOv5、YOLOv7、YOLOv8、YOLOv9等算法的訓(xùn)練免去整理標(biāo)注和配置路徑的繁瑣步驟。壓縮包共兩千個文件其中一千七百個txt標(biāo)簽提供檢測框信息一百七十一個Python腳本覆蓋訓(xùn)練、推理與界面邏輯五十六個yaml文件管理數(shù)據(jù)路徑與參數(shù)另有部分jpg示例圖和pt權(quán)重文件整體約一百八十二兆字節(jié)目錄清晰便于調(diào)用。已有四百三十六人學(xué)習(xí)下載。借助該資源可快速實現(xiàn)玉米葉病害檢測的完整流程覆蓋灰葉斑病、葉枯病、普通銹病和健康葉片四個類別同時PyQt界面支持加載權(quán)重并對本地圖像進(jìn)行可視化識別適合農(nóng)業(yè)AI開發(fā)者、學(xué)生及需要快速落地檢測系統(tǒng)的工程師使用既能學(xué)習(xí)目標(biāo)檢測原理也能直接用于實際項目。1. 拿到Y(jié)OLOv8玉米葉病害這套資源先別急著跑訓(xùn)練玉米葉病害檢測這個方向市面上公開能用的數(shù)據(jù)集本來就少能湊到1500張標(biāo)注圖、附帶可復(fù)現(xiàn)權(quán)重的項目更是稀缺資源。很多剛?cè)腴T的朋友一拿到“yolov8玉米葉病害檢測權(quán)重pyqt界面1500數(shù)據(jù)集”這種包第一反應(yīng)是直接把權(quán)重丟進(jìn)模型里跑預(yù)測結(jié)果發(fā)現(xiàn)界面打不開、權(quán)重版本對不上、數(shù)據(jù)集標(biāo)簽格式不是YOLO需要的txt——這些都是最常見的翻車現(xiàn)場。我個人的建議是先別急著看界面代碼把這條鏈路拆成三個獨立環(huán)節(jié)來理解——數(shù)據(jù)長什么樣、權(quán)重怎么來的、界面怎么調(diào)模型。YOLOv8的檢測流程并不復(fù)雜真正勸退新手的往往是環(huán)境版本錯亂和路徑寫死。這篇文章會從訓(xùn)練環(huán)境搭建、數(shù)據(jù)集處理、權(quán)重訓(xùn)練到PyQt界面封裝把每一步的實用參數(shù)和踩坑經(jīng)驗都過一遍讓新手能復(fù)現(xiàn)、熟手能看出邊界。2. 先搭YOLOv8訓(xùn)練環(huán)境CPU版不用慌GPU版也就三步2.1 環(huán)境版本選型為什么我建議用YOLOv8而不是YOLOv5或v9YOLOv8在目標(biāo)檢測里屬于“開箱即用、改動友好”的版本Ultralytics官方把訓(xùn)練、驗證、導(dǎo)出推理封裝成了統(tǒng)一的Python接口對玉米葉病害這種單類目標(biāo)或者多類病害的檢測任務(wù)非常合適。YOLOv5雖然生態(tài)成熟但anchor-based的設(shè)計在病斑這種小目標(biāo)上不如v8的anchor-free頭靈活YOLOv9、v10的改進(jìn)點主要在大模型和復(fù)雜場景對1500張圖的農(nóng)業(yè)病害數(shù)據(jù)集來說收益不大反而參數(shù)更多、更容易過擬合。病害檢測有個明顯的特點病斑大小差異極大有的孢子堆只有十幾個像素有的枯斑能占滿整片葉子。YOLOv8的C2f結(jié)構(gòu)和解耦檢測頭對小目標(biāo)召回率明顯好于老版本這就是我推薦它做玉米葉病害的核心理由。2.2 CPU版環(huán)境搭建全過程沒有NVIDIA顯卡也完全能跑通訓(xùn)練和推理只是慢。這里給一套當(dāng)前最穩(wěn)的CPU環(huán)境搭建步驟適合Ubuntu 20.04或Windows 10/11。# 1. 創(chuàng)建獨立的conda環(huán)境避免把系統(tǒng)Python搞亂 conda create -n corn_yolo python3.10 -y conda activate corn_yolo # 2. 安裝PyTorch CPU版務(wù)必走官方源不要用默認(rèn)pip源裝GPU版再強行跑CPU pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu # 3. 安裝Ultralytics庫 pip install ultralytics # 4. 驗證安裝 python -c from ultralytics import YOLO; model YOLO(yolov8n.pt); print(OK)邏輯說明第一步建獨立conda環(huán)境是為了隔離依賴YOLOv8需要torch1.8但如果你系統(tǒng)里已有其他項目用了torch 2.0以上版本直接裝容易把環(huán)境攪亂。第二步指定--index-url參數(shù)是為了確保下載的是CPU版本的torch如果你在無GPU機(jī)器上直接用默認(rèn)源裝pip會默認(rèn)拉一個帶CUDA的大包安裝耗時翻倍而且跑不起來。第三步安裝ultralytics庫這個庫本身就包含YOLOv8的全部模型定義和訓(xùn)練工具。第四步的驗證命令非常關(guān)鍵它會下載一個yolov8n.pt預(yù)訓(xùn)練權(quán)重下載成功后輸出OK才說明環(huán)境鏈路是通的。如果你輸出的是ModuleNotFoundError多半是conda環(huán)境和當(dāng)前終端沒對上用which python確認(rèn)一下路徑。2.3 GPU版環(huán)境的參數(shù)選擇與依賴坑有NVIDIA顯卡的話訓(xùn)練1500張數(shù)據(jù)集的時間能從CPU的十幾小時壓縮到一小時以內(nèi)。GPU版環(huán)境搭建和CPU版唯一的區(qū)別在torch安裝這一步# GPU版安裝CUDA版本要跟顯卡驅(qū)動匹配 nvidia-smi # 先看驅(qū)動的CUDA版本比如11.8或12.1 pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118這里的參數(shù)坑在于--index-url后面的cu118必須對應(yīng)nvidia-smi顯示的CUDA版本。很多人裝了CUDA 12.1的驅(qū)動結(jié)果在Ultralytics加載模型時提示torch.cuda.is_available()返回False查半天發(fā)現(xiàn)是torch裝了cu118版本實際驅(qū)動只認(rèn)cu121的編譯包。對GTX 1660 Ti這種8GB顯存的卡訓(xùn)練yolov8n或yolov8s足夠用batch size可以設(shè)到16如果是老款4GB顯存卡batch size必須降為4-8否則直接爆顯存。提示顯卡驅(qū)動版本和CUDA運行庫是兩回事。nvidia-smi顯示的CUDA版本只是驅(qū)動支持的最高版本不代表你需要裝那個版本的torch低版本的torch CUDA包在高版本驅(qū)動上照常運行。3. 1500張玉米葉數(shù)據(jù)集的處理從標(biāo)注格式到劃分策略3.1 數(shù)據(jù)集原始格式長什么樣怎么檢查玉米葉病害數(shù)據(jù)集通常由仿真的病斑合成圖與真實田間圖混合構(gòu)成。拿到“1500數(shù)據(jù)集”時第一件事不是看圖片而是看標(biāo)注文件的格式。常見兩種情況一種是VOC的xml標(biāo)注一種是YOLO的txt標(biāo)注。YOLOv8只認(rèn)txt格式而且要求每張圖片的txt與jpg同名且放在同一個文件夾下。# 快速檢查1500張數(shù)據(jù)集中標(biāo)注是否完整 cd dataset_dir find . -name *.jpg | wc -l # 統(tǒng)計圖片數(shù) find . -name *.txt | wc -l # 統(tǒng)計標(biāo)注數(shù)如果txt數(shù)量明顯少于jpg數(shù)量說明有漏標(biāo)圖直接拿去訓(xùn)練會報assert no empty images之類的錯誤或者模型默默跳過這些圖片。這種情況需要把漏標(biāo)的圖片單獨拎出來。還有一種常見情況是xml格式——也就是VOC格式——需要先用腳本轉(zhuǎn)換。轉(zhuǎn)換腳本的核心就是把xml里的bndbox坐標(biāo)值除以圖片寬高歸一化到0-1之間再按“類別id 中心x 中心y 寬 高”的格式寫入txt。特別提醒類別id要從0開始計數(shù)這個坑我見很多人跳過導(dǎo)致訓(xùn)練時類的對應(yīng)關(guān)系全錯。3.2 數(shù)據(jù)集劃分固定隨機(jī)種子別讓結(jié)果不可復(fù)現(xiàn)數(shù)據(jù)集劃分方式直接決定了訓(xùn)練效果能不能復(fù)現(xiàn)。很多人用Ultralytics的默認(rèn)機(jī)制它會自動把數(shù)據(jù)按比例劃分成訓(xùn)練集和驗證集但這種劃分方式每次跑的結(jié)果不一定一致。正確做法是手動劃分并固定隨機(jī)種子這樣你復(fù)現(xiàn)出來的結(jié)果才是穩(wěn)定的。import os import random import shutil from sklearn.model_selection import train_test_split # 固定隨機(jī)種子保證每次劃分結(jié)果一致 random.seed(42) img_paths [f for f in os.listdir(images) if f.endswith(.jpg)] train_files, val_files train_test_split(img_paths, test_size0.2, random_state42) # 按YOLO約定目錄結(jié)構(gòu)組織 for split, files in [(train, train_files), (val, val_files)]: os.makedirs(fdatasets/corn/{split}/images, exist_okTrue) os.makedirs(fdatasets/corn/{split}/labels, exist_okTrue) for fname in files: shutil.copy(fimages/{fname}, fdatasets/corn/{split}/images/{fname}) label fname.replace(.jpg, .txt) if os.path.exists(flabels/{label}): shutil.copy(flabels/{label}, fdatasets/corn/{split}/labels/{label})這段代碼的邏輯先把所有圖片路徑讀進(jìn)來用train_test_split按80%/20%劃分再把文件和對應(yīng)標(biāo)簽復(fù)制到Y(jié)OLO規(guī)定的目錄結(jié)構(gòu)下。參數(shù)說明test_size0.2表示1500張圖里大約300張留給驗證集random_state42是關(guān)鍵——換成別的值劃分結(jié)果就不同最后的模型指標(biāo)也不同論文復(fù)現(xiàn)時千萬不能改。提示圖片增強時不要用隨機(jī)翻轉(zhuǎn)和旋轉(zhuǎn)玉米葉病害的左右對稱性遠(yuǎn)不如通用物體檢測翻轉(zhuǎn)后病斑的位置和形態(tài)容易出現(xiàn)不符合現(xiàn)實的樣本降低驗證集的可信度。4. 訓(xùn)練YOLOv8玉米葉病害權(quán)重核心參數(shù)與損失曲線判讀4.1 寫一份適合1500張數(shù)據(jù)集的訓(xùn)練配置# corn_dataset.yaml train: datasets/corn/train/images val: datasets/corn/val/images nc: 4 names: [灰斑病, 銹病, 大斑病, 健康]這里nc: 4表示檢測4個類別如果病害種類不同按自己的標(biāo)注實際數(shù)量改。names列表的順序必須和訓(xùn)練時類別id的對應(yīng)關(guān)系一致不然預(yù)測結(jié)果會張冠李戴。4.2 訓(xùn)練命令的關(guān)鍵參數(shù)含義yolo detect train \ modelyolov8n.pt \ datacorn_dataset.yaml \ epochs100 \ imgsz640 \ batch16 \ patience20 \ projectcorn_project \ nameexp1 \ device0 \ lr00.01參數(shù)說明modelyolov8n.pt代表從COCO預(yù)訓(xùn)練權(quán)重開始微調(diào)比從零訓(xùn)練收斂快很多這是遷移學(xué)習(xí)的核心用法。epochs100對1500張數(shù)據(jù)來說夠用了再多就有過擬合風(fēng)險。imgsz640是輸入分辨率玉米葉病斑小的可以用640x640如果都是大斑塊用512能顯著提速。patience20是早停耐心值——驗證集指標(biāo)連續(xù)20個epoch不提升就自動?？梢苑乐篃o效訓(xùn)練。batch16是顯存允許范圍內(nèi)的值GTX 1660 Ti可跑16GB顯存可以調(diào)到32。如果目標(biāo)是在PyQt界面里實時檢測建議顯存夠的情況下把imgsz控制在640以內(nèi)因為推理速度跟輸入尺寸直接相關(guān)。訓(xùn)練完成后runs/detect/exp1/weights/下會出現(xiàn)best.pt和last.pt兩個權(quán)重文件best.pt是按驗證集指標(biāo)挑出的最優(yōu)模型界面里應(yīng)該加載best.pt而不是last.pt這一點常被忽略導(dǎo)致部署的模型比訓(xùn)練評估時差一截。4.3 損失函數(shù)曲線怎么看拐點意味著什么訓(xùn)練完成后在runs/detect/exp1/目錄下會生成results.png里面畫了box_loss、cls_loss、dfl_loss和mAP曲線的變化趨勢。我一般只看三條train/box_loss訓(xùn)練集上的回歸損失趨勢應(yīng)該是收斂下降的如果出現(xiàn)先降后升說明學(xué)習(xí)率太大或數(shù)據(jù)集存在噪聲標(biāo)注。val/box_loss驗證集上的回歸損失這個值突然反彈是過擬合的早期信號對應(yīng)訓(xùn)練里應(yīng)該早停。metrics/mAP50-95綜合指標(biāo)玉米葉病害這類目標(biāo)不算難檢測如果最終mAP50在0.85以上算合格低于0.7說明數(shù)據(jù)標(biāo)注有問題或類別數(shù)配置錯了。畫損失函數(shù)曲線還有一個更直接的腳本import pandas as pd import matplotlib.pyplot as plt # 從訓(xùn)練日志中讀取指標(biāo) df pd.read_csv(runs/detect/exp1/results.csv) plt.plot(df[epoch], df[train/box_loss], labeltrain box_loss) plt.plot(df[epoch], df[val/box_loss], labelval box_loss) plt.xlabel(epoch) plt.ylabel(loss) plt.legend() plt.savefig(loss_curve.png)5. 用PyQt做玉米葉病害檢測界面加載權(quán)重與實時推理5.1 環(huán)境準(zhǔn)備與界面功能拆解PyQt界面在整個項目里承擔(dān)的職責(zé)很清晰加載best.pt權(quán)重讀取圖片或調(diào)起攝像頭把模型推理結(jié)果繪制在界面上。界面代碼基于PyQt5的QMainWindow框架核心組件包括圖片顯示區(qū)、按鈕組、結(jié)果文本區(qū)。這是最常見也最穩(wěn)妥的布局——不需要復(fù)雜的自定義控件入門者在網(wǎng)上找到的絕大多數(shù)“YOLOv8PyQt”界面模板都是這個結(jié)構(gòu)。5.2 用PyQt實現(xiàn)檢測的核心代碼import sys from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QPushButton, QVBoxLayout, QWidget, QFileDialog from PyQt5.QtGui import QPixmap, QImage from PyQt5.QtCore import Qt import cv2 from ultralytics import YOLO class CornDetectWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle(玉米葉病害檢測) self.model YOLO(best.pt) # 加載訓(xùn)練好的權(quán)重 self.image_label QLabel(self) self.image_label.setAlignment(Qt.AlignCenter) self.btn QPushButton(選擇圖片, self) self.btn.clicked.connect(self.detect_image) layout QVBoxLayout() layout.addWidget(self.image_label) layout.addWidget(self.btn) container QWidget() container.setLayout(layout) self.setCentralWidget(container) self.resize(960, 720) def detect_image(self): path, _ QFileDialog.getOpenFileName(self, 選擇圖片, , Images(*.jpg *.png)) if not path: return results self.model.predict(path, conf0.4) # 讀取原始圖像并繪制檢測框 img cv2.imread(path) for r in results: boxes r.boxes.xyxy.cpu().numpy() confs r.boxes.conf.cpu().numpy() for box, conf in zip(boxes, confs): x1, y1, x2, y2 map(int, box) cv2.rectangle(img, (x1, y1), (x2, y2), (0, 255, 0), 2) cv2.putText(img, f{conf:.2f}, (x1, y1-10), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2) # 把OpenCV BGR圖像轉(zhuǎn)為Qt RGB顯示 img_rgb cv2.cvtColor(img, cv2.COLOR_BGR2RGB) h, w, ch img_rgb.shape bytes_per_line ch * w qt_img QImage(img_rgb.data, w, h, bytes_per_line, QImage.Format_RGB888) self.image_label.setPixmap(QPixmap.fromImage(qt_img).scaled( self.image_label.width(), self.image_label.height(), Qt.KeepAspectRatio)) if __name__ __main__: app QApplication(sys.argv) win CornDetectWindow() win.show() sys.exit(app.exec_())關(guān)鍵參數(shù)與邏輯說明self.model YOLO(best.pt)必須確保best.pt和這個.py文件在同一目錄否則要用絕對路徑。self.model.predict(path, conf0.4)中的conf0.4是置信度閾值低于0.4的預(yù)測框會被丟棄如果你感覺漏檢多降到0.25如果你覺得誤檢多提到0.5。這個值是界面調(diào)優(yōu)時最需要反復(fù)測的參數(shù)沒有絕對正確的默認(rèn)值靠的是對驗證集結(jié)果的人工觀察。注意代碼里cv2.rectangle繪制的是原始圖像results給出的坐標(biāo)是640x640輸入尺寸對應(yīng)的而cv2.imread讀出來的是原圖尺寸但實際上YOLOv8內(nèi)部已經(jīng)做了等比縮放映射輸出的xyxy坐標(biāo)就是原圖坐標(biāo)系下的值不需要手工換算。5.3 攝像頭實時檢測的參數(shù)調(diào)整如果要做實時檢測比如田間用攝像頭掃葉片前面的檢測單張圖代碼要換成循環(huán)讀幀cap cv2.VideoCapture(0) while True: ret, frame cap.read() if not ret: break results self.model.predict(frame, imgsz640, conf0.4, verboseFalse) # 繪制與display邏輯同上 if cv2.waitKey(1) 0xFF ord(q): break這里的imgsz640和verboseFalse很關(guān)鍵。前者控制推理分辨率顯存不夠或CPU推理時降為320能把幀率從5fps提升到10fps以上代價是小目標(biāo)檢測能力下降后者關(guān)閉日志輸出否則終端每幀都會刷一條推理信息界面卡頓明顯。6. 這個項目最常見的坑與解決記錄6.1 權(quán)重文件加載報錯“cannot be mapped to a model”現(xiàn)象YOLO(best.pt)拋出RuntimeError提示模型結(jié)構(gòu)不匹配。原因訓(xùn)練時用的YOLOv8版本和界面推理時用的ultralytics版本不一致v8早期版本和后期版本對C2f模塊的命名有差異。另一個常見原因是加載了last.pt但訓(xùn)練中斷導(dǎo)致文件損壞。解決在訓(xùn)練和部署環(huán)境里統(tǒng)一版本號訓(xùn)練時如果用的pip install ultralytics8.2.0部署環(huán)境也裝同一版本。如果文件損壞重新訓(xùn)練或回退到best.pt。6.2 PyQt界面打開后無響應(yīng)或黑屏現(xiàn)象點擊“選擇圖片”后界面卡住或顯示全黑。原因在UI線程里直接調(diào)用model.predict()推理過程阻塞了事件循環(huán)。圖片分辨率大時推理耗時數(shù)秒整個窗口會處于“未響應(yīng)”狀態(tài)。解決把推理放到QThread里主線程只負(fù)責(zé)刷新界面。最小改法是給predict傳入devicecpu以外的優(yōu)化參數(shù)比如halfTrue半精度推理。對于新手最簡單的做法是把predict的imgsz調(diào)低如320縮短阻塞時間本質(zhì)不等于解決但對學(xué)習(xí)夠用。6.3 訓(xùn)練時顯存不足OOM現(xiàn)象epoch剛開始就報CUDA out of memory然后進(jìn)程被kill。原因batch size太大或者imgsz太大或訓(xùn)練時同時開了驗證驗證階段也占顯存。玉米葉病害的imgsz通常用640已經(jīng)偏大此時病斑很小放大到640沒有必要。解決把batch降到4或8把imgsz改為480在訓(xùn)練參數(shù)里加workers0Windows上worker多會復(fù)制顯存。這三個調(diào)整能解決九成OOM問題。6.4 攝像頭檢測時OpenCV與PyQt的通道順序不對現(xiàn)象界面里葉片顏色偏藍(lán)或偏紅檢測框位置正確但顏色詭異。原因OpenCV默認(rèn)讀入BGR格式QImage構(gòu)造函數(shù)默認(rèn)按RGB解析。代碼漏了cv2.cvtColor(img, cv2.COLOR_BGR2RGB)這一步就會出現(xiàn)偏色而網(wǎng)絡(luò)推理本身不受影響因為YOLOv8在預(yù)處理階段做了BGR轉(zhuǎn)RGB。解決仔細(xì)核對5.2節(jié)代碼中顏色轉(zhuǎn)換的步驟偏色問題百分之百出在這行。檢測框繪制要基于BGR的數(shù)據(jù)顯示前轉(zhuǎn)一次RGB順序不能反。注意以上四類坑里版本不一致是出現(xiàn)頻率最高的。換環(huán)境時不要隨手pip install -U ultralytics這會把權(quán)重文件可加載性全部打亂。建議訓(xùn)練和部署共用同一臺機(jī)器或同一個conda環(huán)境。7. 讓玉米葉病害系統(tǒng)更可靠的三個進(jìn)階做法項目能跑通只是及格線面向?qū)嶋H使用還需要補三個細(xì)節(jié)。第一個是在界面里加入檢測結(jié)果置信度分布顯示——雖然這個項目沒有要求但我一直習(xí)慣把每幀預(yù)測的conf值打印或存日志積累二三十張圖后就能看出conf0.4這個閾值設(shè)得是否合理。閾值不合適時會出現(xiàn)大量0.38-0.45之間搖擺的檢測框這時候?qū)幙砂验撝堤岣叩?.5也不要讓它反復(fù)閃爍。第二個是權(quán)重文件的管理習(xí)慣。訓(xùn)練完的best.pt不要直接覆蓋每次調(diào)參前把它改名保存為corn_v1_best.pt、corn_v2_best.pt。我吃過一次虧調(diào)參后新權(quán)重mAP反而下降想回滾卻發(fā)現(xiàn)舊權(quán)重被覆蓋了只能重新訓(xùn)練兩小時。給權(quán)重加版本后綴這個習(xí)慣能給你買一張后悔藥。第三個是驗證集的留存不可省略。1500張數(shù)據(jù)集里如果包含仿真合成圖片模型在合成圖上可能表現(xiàn)非常好到了真實田間照片上卻一塌糊涂。我會在訓(xùn)練前單獨挑出50張真實拍攝的葉片圖不進(jìn)訓(xùn)練集也不進(jìn)驗證集訓(xùn)練完專門跑一遍這批“留出集”看mAP掉得多不多。如果掉得厲害說明合成圖與真實圖仍有域差異——這時候界面里應(yīng)該加一個“增強對比度”的預(yù)處理選項很多病害檢測項目靠這步就能把真實場景的漏檢率降一半。我現(xiàn)在的習(xí)慣是每次訓(xùn)練完先看損失曲線是否收斂再跑留出集最后才打包給界面用。這套流程下來踩坑的次數(shù)少了很多。希望幫到你。本文還有配套的精品資源點擊獲取