博客 / 詳情

返回

基於YOLOv8的牛行為檢測識別項目|完整源碼數據集+PyQt5界面+完整訓練流程+開箱即用!

基於YOLOv8的牛行為檢測識別項目|完整源碼數據集+PyQt5界面+完整訓練流程+開箱即用!

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

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

本系統通過 PyQt5 圖形界面 提供多種輸入方式,包括:

  • 圖片識別:單張或批量圖片檢測牛的行為狀態。
  • 文件夾識別:批量處理指定文件夾內的圖片。
  • 視頻識別:實時檢測視頻中的牛行為。
  • 攝像頭識別:實時監控牛的行為,適合牧場和飼養場應用。

檢測結果可直接在界面顯示,包括行為類別和置信度,並支持保存檢測結果圖像或視頻。

項目摘要

本項目集成 YOLOv8 行為檢測模型PyQt5 圖形界面工具,實現牛行為的自動識別與分類。特點包括:

  • 開箱即用:無需複雜配置,下載源碼即可運行。
  • 多場景支持:適用於牧場、養殖場、實驗環境等。
  • 可擴展性強:可在現有模型基礎上添加新行為類別。
  • 實用性高:可輔助牛羣管理、健康監控和行為分析。

前言

隨着智能農業的發展,自動化行為識別在畜牧管理中發揮着越來越重要的作用。傳統人工觀察效率低,難以覆蓋大規模養殖場。而通過 YOLOv8 目標檢測模型 配合 PyQt5 界面工具,可以實現實時、準確的牛行為檢測,幫助農場管理者瞭解牛羣健康狀態、監測異常行為,提高生產效率。

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

1. 牛行為識別模型

  • 使用 YOLOv8 作為基礎檢測模型。
  • 對牛卧、牛站立、牛行走三種行為進行檢測。
  • 訓練數據豐富,模型精度高,可適應不同場景。

2. PyQt5 圖形界面

  • 提供操作簡單的界面,無需命令行操作。
  • 支持多種輸入方式:圖片、文件夾、視頻、攝像頭。
  • 實時顯示檢測結果,支持結果導出。

3. 模型訓練與優化

  • 提供完整訓練流程,包括數據準備、模型訓練、驗證及權重保存。
  • 支持模型參數調整,如學習率、批大小、訓練輪數等。
  • 可對新場景進行遷移訓練,提高模型在特定環境下的表現。

4. 部署與使用

  • 下載源碼與數據集後即可運行,無需額外依賴複雜環境。
  • 可用於科研、教育及智能農業等應用場景。
  • 支持後續擴展,如增加更多牛行為類別或結合行為分析系統。

二、軟件效果演示

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

(1)單圖片檢測演示

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

image-20251109013000270


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

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

image-20251109013021157


(3)視頻檢測演示

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

image-20251109013042871


(4)攝像頭檢測演示

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

image-20251109013107663


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

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

image-20251109013138097

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

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-20251109013251951

image-20251109013236807

image-20251109013224331

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-20251109013201383

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-20251109013317729

四.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/BV11FkiB7ENw

image-20250801135823301

包含:

📦完整項目源碼

📦 預訓練模型權重

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

總結

本項目通過 YOLOv8 模型與 PyQt5 界面結合,實現了牛行為的高效識別與分類。5000張高質量標註數據保證了模型的準確性,多樣化場景增強了泛化能力。系統簡單易用,支持圖片、視頻、攝像頭多種輸入方式,為智能養殖和畜牧管理提供了高效工具。無論是科研實驗還是實際牧場監控,本項目都可快速部署,開箱即用。

user avatar codingdgsun 頭像 mo_or 頭像 redorblack 頭像 knifeblade 頭像 qiehxb8 頭像 luoshenshen 頭像 jacheut 頭像 mackyhuang 頭像 tigerb 頭像 u_16099188 頭像 skysailstar 頭像 hsiang 頭像
17 位用戶收藏了這個故事!

發佈 評論

Some HTML is okay.