Stories

Detail Return Return

基於YOLOv8的番茄葉片病害識別系統|完整源碼數據集+PyQt5界面+完整訓練流程+開箱即用! - Stories Detail

基於YOLOv8的番茄葉片病害識別系統|完整源碼數據集+PyQt5界面+完整訓練流程+開箱即用!

源碼包含:完整YOLOv8訓練代碼+數據集(帶標註)+權重文件+直接可允許檢測的yolo檢測程序+直接部署教程/訓練教程

源碼在文末嗶哩嗶哩視頻簡介處獲取。

基本功能演示

支持 圖片檢測、視頻檢測、實時攝像頭檢測

支持 界面一鍵加載模型與快速預測

支持 目標框 + 類別名稱 + 置信度可視化顯示

支持 批量檢測並自動保存檢測結果

支持 模型自定義替換與快速重新訓練

項目摘要

本項目針對番茄葉片常見病害難以人工快速識別的問題,構建了一個基於 YOLOv8 的智能病害識別系統。系統利用深度學習模型對葉片圖像進行檢測與分類,可自動分辨細菌、真菌、病毒引起的多種病害類型,並支持實時檢測與可視化展示。通過結合 PyQt5 圖形化界面,系統可直接作為農業大棚監測、農技人員巡檢、科研教學示範等場景的 開箱即用智能工具

基於YOLOv8的番茄葉片病害識別系統,可檢測下面10種疾病

'番茄細菌性斑點病',
'番茄早疫病',
'番茄晚疫病',
'番茄葉黴病',
'番茄褐斑病',
'番茄紅蜘蛛(雙斑葉蟎)',
'番茄靶斑病',
'番茄黃化卷葉病毒病',
'健康番茄',
'番茄花葉病毒病'

前言

番茄是全球重要的經濟作物之一,但其葉片病害類型多、病原複雜、症狀相似,傳統依賴人工診斷經驗,存在 效率低、識別難、延誤最佳防治時機 等問題。隨着深度學習的發展,基於圖像的病害識別技術已經能夠達到穩定高效的識別精度。

本項目基於 YOLOv8 目標檢測模型,在 帶標籤葉片圖像 上進行了系統訓練與優化,並結合 PyQt5 實現了界面化部署,使得 非 AI 技術人員也可直接使用,在農業生產中具有較高落地價值。

一、軟件核心功能介紹及效果演示

功能模塊 描述
病害自動檢測 支持對番茄葉片圖像進行多類別病害識別,並輸出目標框和類別標籤
實時視頻/攝像頭識別 支持 USB 攝像頭、工控機相機或本地視頻流實時識別
批量圖片檢測 支持選擇文件夾一鍵處理並保存結果
PyQt5 圖形界面 支持無需命令行,用户可直接操作
模型可替換 / 可再訓練 支持更換權重、重新訓練、自定義數據擴展
自動保存結果 檢測標註框、預測數據與結果圖自動保存

二、軟件效果演示

為了直觀展示本系統基於 YOLOv8 模型的檢測能力,我們設計了多種操作場景,涵蓋靜態圖片、批量圖片、視頻以及實時攝像頭流的檢測演示。

(1)單圖片檢測演示

用户點擊“選擇圖片”,即可加載本地圖像並執行檢測:

image-20251107205716823


(2)多文件夾圖片檢測演示

用户可選擇包含多張圖像的文件夾,系統會批量檢測並生成結果圖。

image-20251107205528296


(3)視頻檢測演示

支持上傳視頻文件,系統會逐幀處理並生成目標檢測結果,可選保存輸出視頻:

image-20251107205814529


(4)攝像頭檢測演示

實時檢測是系統中的核心應用之一,系統可直接調用攝像頭進行檢測。由於原理和視頻檢測相同,就不重複演示了。


(5)保存圖片與視頻檢測結果

用户可通過按鈕勾選是否保存檢測結果,所有檢測圖像自動加框標註並保存至指定文件夾,支持後續數據分析與複審。

image-20251107205836257

三、模型的訓練、評估與推理

YOLOv8是Ultralytics公司發佈的新一代目標檢測模型,採用更輕量的架構、更先進的損失函數(如CIoU、TaskAlignedAssigner)與Anchor-Free策略,在COCO等數據集上表現優異。
其核心優勢如下:

  • 高速推理,適合實時檢測任務
  • 支持Anchor-Free檢測
  • 支持可擴展的Backbone和Neck結構
  • 原生支持ONNX導出與部署

3.1 YOLOv8的基本原理

YOLOv8 是 Ultralytics 發佈的新一代實時目標檢測模型,具備如下優勢:

  • 速度快:推理速度提升明顯;
  • 準確率高:支持 Anchor-Free 架構;
  • 支持分類/檢測/分割/姿態多任務
  • 本項目使用 YOLOv8 的 Detection 分支,訓練時每類表情均標註為獨立目標。

YOLOv8 由Ultralytics 於 2023 年 1 月 10 日發佈,在準確性和速度方面具有尖端性能。在以往YOLO 版本的基礎上,YOLOv8 引入了新的功能和優化,使其成為廣泛應用中各種物體檢測任務的理想選擇。

image-20250526165954475

YOLOv8原理圖如下:

image-20250526170118103

3.2 數據集準備與訓練

採用 YOLO 格式的數據集結構如下:

dataset/
├── images/
│   ├── train/
│   └── val/
├── labels/
│   ├── train/
│   └── val/

每張圖像有對應的 .txt 文件,內容格式為:

4 0.5096721233576642 0.352838390077821 0.3947600423357664 0.31825755058365757

分類包括(可自定義):

image-20251107210033613

3.3. 訓練結果評估

訓練完成後,將在 runs/detect/train 目錄生成結果文件,包括:

  • results.png:損失曲線和 mAP 曲線;
  • weights/best.pt:最佳模型權重;
  • confusion_matrix.png:混淆矩陣分析圖。
若 mAP@0.5 達到 90% 以上,即可用於部署。

在深度學習領域,我們通常通過觀察損失函數下降的曲線來評估模型的訓練狀態。YOLOv8訓練過程中,主要包含三種損失:定位損失(box_loss)、分類損失(cls_loss)和動態特徵損失(dfl_loss)。訓練完成後,相關的訓練記錄和結果文件會保存在runs/目錄下,具體內容如下:

image-20251107210012432

3.4檢測結果識別

使用 PyTorch 推理接口加載模型:

import cv2
from ultralytics import YOLO
import torch
from torch.serialization import safe_globals
from ultralytics.nn.tasks import DetectionModel

# 加入可信模型結構
safe_globals().add(DetectionModel)

# 加載模型並推理
model = YOLO('runs/detect/train/weights/best.pt')
results = model('test.jpg', save=True, conf=0.25)

# 獲取保存後的圖像路徑
# 默認保存到 runs/detect/predict/ 目錄
save_path = results[0].save_dir / results[0].path.name

# 使用 OpenCV 加載並顯示圖像
img = cv2.imread(str(save_path))
cv2.imshow('Detection Result', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

預測結果包含類別、置信度、邊框座標等信息。

image-20251107210054183

image-20251107210111005

四.YOLOV8+YOLOUI完整源碼打包

本文涉及到的完整全部程序文件:包括python源碼、數據集、訓練代碼、UI文件、測試圖片視頻等(見下圖),獲取方式見【4.2 完整源碼下載】:

4.1 項目開箱即用

作者已將整個工程打包。包含已訓練完成的權重,讀者可不用自行訓練直接運行檢測。

運行項目只需輸入下面命令。

python main.py

讀者也可自行配置訓練集,或使用打包好的數據集直接訓練。

自行訓練項目只需輸入下面命令。

yolo detect train data=datasets/expression/loopy.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 batch=16 lr0=0.001

4.2 完整源碼

至項目實錄視頻下方獲取:

https://www.bilibili.com/video/BV1P111BJEjk/

image-20250801135823301

包含:

📦完整項目源碼

📦 預訓練模型權重

🗂️ 數據集地址(含標註腳本)

總結

本項目基於 YOLOv8 模型構建了一個 可直接落地使用的番茄葉片病害智能識別系統。通過對大規模番茄病害圖像進行標註與訓練,系統能夠精準識別包括細菌、真菌、病毒等在內的多種典型葉片病害類型,並支持 圖片、視頻、攝像頭實時檢測與批量處理。同時,結合 PyQt5 圖形化界面實現了真正的 “零代碼上手”,即使沒有深度學習經驗的農業從業者、農技人員或科研人員也能輕鬆部署與使用。

相比傳統人工巡檢方式,本系統在 識別效率、準確性、可擴展性與自動化能力 方面都具有明顯優勢,能夠有效減少誤診漏診、提升病害防治時效,有助於農業生產過程向 智能化、精準化、現代化 方向發展。未來,本系統可進一步拓展到更多作物病害場景,或結合無人機、智能大棚等設備實現自動化監控,為數字農業提供更加完整的智能解決方案。

user avatar u_16776161 Avatar u_17513518 Avatar u_15511034 Avatar aitinggedejinzhengu Avatar youqingyouyidejidan Avatar
Favorites 5 users favorite the story!
Favorites

Add a new Comments

Some HTML is okay.