識(shí)別系統(tǒng)設(shè)計(jì)與實(shí)現(xiàn))
這次我們來(lái)看一個(gè)把目標(biāo)檢測(cè)算法和桌面端工具結(jié)合得很典型的項(xiàng)目基于 YOLOv8 PyQt5 的麥穗稻穗檢測(cè)識(shí)別系統(tǒng)。這個(gè)項(xiàng)目本身不是新概念但它的價(jià)值在于落地形態(tài)很完整。YOLOv8 負(fù)責(zé)核心的麥穗稻穗目標(biāo)檢測(cè)PyQt5 負(fù)責(zé)提供可視化的桌面交互界面二者通過(guò) Python 生態(tài)整合最終產(chǎn)出的是一個(gè)可以獨(dú)立運(yùn)行的桌面應(yīng)用程序。相比單純跑一段檢測(cè)腳本這種帶圖形界面的系統(tǒng)更適合農(nóng)業(yè)科研場(chǎng)景中的樣本統(tǒng)計(jì)、田間調(diào)查記錄也非常適合作為畢業(yè)設(shè)計(jì)、課程項(xiàng)目或者工業(yè)視覺(jué)的參考原型。這篇文章會(huì)重點(diǎn)覆蓋這幾個(gè)部分YOLOv8 模型如何準(zhǔn)備和訓(xùn)練、PyQt5 桌面界面如何搭建、檢測(cè)結(jié)果如何在界面上實(shí)時(shí)展示、攝像頭/圖片/視頻三種輸入模式怎么切換、界面與推理線程怎么解耦避免卡頓以及最終如何打包成可分發(fā)的一鍵運(yùn)行程序。如果你正在做類似的檢測(cè)識(shí)別系統(tǒng)這篇文章可以直接參考整體架構(gòu)和關(guān)鍵代碼。1. 核心能力速覽能力項(xiàng)說(shuō)明項(xiàng)目類型桌面端目標(biāo)檢測(cè)識(shí)別系統(tǒng)檢測(cè)算法YOLOv8Ultralytics YOLOv8界面框架PyQt5Python 5檢測(cè)目標(biāo)麥穗、稻穗可通過(guò)數(shù)據(jù)集擴(kuò)展輸入方式本地圖片、視頻文件、攝像頭實(shí)時(shí)畫(huà)面輸出內(nèi)容檢測(cè)框、類別標(biāo)簽、置信度、目標(biāo)數(shù)量統(tǒng)計(jì)啟動(dòng)方式Python 腳本啟動(dòng) / PyInstaller 打包為 exeGPU 支持支持 CUDA 加速也支持 CPU 推理顯存需求需按模型版本和輸入分辨率測(cè)試YOLOv8n 在 CPU 上也能運(yùn)行API 接口非本項(xiàng)目核心但 YOLOv8 檢測(cè)邏輯可獨(dú)立封裝批量任務(wù)可擴(kuò)展界面邏輯和推理邏輯分離后可接入批量目錄處理適合場(chǎng)景農(nóng)業(yè)科研統(tǒng)計(jì)、田間調(diào)查、教學(xué)演示、畢業(yè)設(shè)計(jì)、算法原型驗(yàn)證從項(xiàng)目定位來(lái)看這個(gè)系統(tǒng)并不追求在算法層面做出創(chuàng)新而是把 YOLOv8 的檢測(cè)能力和 PyQt5 的桌面交互能力打通。它解決的核心問(wèn)題是讓沒(méi)有命令行操作習(xí)慣的使用者也能通過(guò)鼠標(biāo)點(diǎn)擊完成目標(biāo)檢測(cè)和結(jié)果統(tǒng)計(jì)。2. 適用場(chǎng)景與使用邊界2.1 適合誰(shuí)用農(nóng)業(yè)科研人員、植保相關(guān)專業(yè)的學(xué)生、需要做作物生長(zhǎng)情況統(tǒng)計(jì)的研究團(tuán)隊(duì)是這個(gè)系統(tǒng)最直接的受眾。比如在田間拍攝一批麥穗圖片過(guò)去靠人工數(shù)穗數(shù)費(fèi)時(shí)費(fèi)力現(xiàn)在可以通過(guò)模型自動(dòng)標(biāo)注每個(gè)麥穗的位置并輸出數(shù)量統(tǒng)計(jì)。對(duì)于計(jì)算機(jī)視覺(jué)方向的學(xué)生來(lái)說(shuō)這個(gè)項(xiàng)目的工程價(jià)值更明顯。它演示了如何把 YOLOv8 模型封裝成可交互的桌面應(yīng)用這套架構(gòu)可以遷移到其他檢測(cè)任務(wù)中比如果實(shí)計(jì)數(shù)、工廠零件檢測(cè)、安防目標(biāo)識(shí)別等。2.2 能解決什么問(wèn)題檢測(cè)識(shí)別系統(tǒng)要解決的并不是“識(shí)別得有多準(zhǔn)”這一個(gè)問(wèn)題而是“從拿到圖像到得到結(jié)果”這個(gè)完整鏈路的問(wèn)題。具體包括圖片預(yù)覽和檢測(cè)結(jié)果疊加顯示檢測(cè)框坐標(biāo)、類別、置信度的讀取和保存單張圖片中目標(biāo)數(shù)量的自動(dòng)統(tǒng)計(jì)視頻流和攝像頭畫(huà)面中的連續(xù)目標(biāo)檢測(cè)檢測(cè)結(jié)果的導(dǎo)出方便后續(xù)數(shù)據(jù)分析2.3 不適合什么場(chǎng)景實(shí)時(shí)性要求特別高的場(chǎng)景不適合直接用 PyQt5 的簡(jiǎn)單方案界面渲染本身會(huì)占用一定資源嵌入式設(shè)備部署也不是這個(gè)項(xiàng)目的強(qiáng)項(xiàng)更適合用 YOLOv8 導(dǎo)出 ONNX 或 TensorRT 后在 RK3588、Jetson 等邊緣設(shè)備上單獨(dú)部署大規(guī)模并行推理服務(wù)場(chǎng)景則建議把檢測(cè)邏輯獨(dú)立成 API 服務(wù)而不是依賴桌面界面。2.4 版權(quán)與授權(quán)邊界使用 YOLOv8 開(kāi)源框架時(shí)需要注意 Ultralytics 的許可證約定。訓(xùn)練數(shù)據(jù)如果是自己采集的田間圖片需要確保拍攝對(duì)象不涉及他人隱私或保密區(qū)域。如果使用公開(kāi)數(shù)據(jù)集則要遵循數(shù)據(jù)集的授權(quán)協(xié)議。系統(tǒng)如果涉及商業(yè)交付需要確認(rèn)模型權(quán)重、數(shù)據(jù)集和代碼部分的許可兼容性。3. 環(huán)境準(zhǔn)備與前置條件3.1 硬件環(huán)境YOLOv8 官方提供多種尺寸的模型n、s、m、l、x。其中 YOLOv8n 最輕量CPU 也能推理YOLOv8s 和 YOLOv8m 在普通消費(fèi)級(jí)顯卡上有比較好的精度和速度平衡。如果手頭只有核顯或者無(wú)獨(dú)顯設(shè)備優(yōu)先選 YOLOv8n。如果使用 NVIDIA 顯卡并希望通過(guò) CUDA 加速需要提前裝好顯卡驅(qū)動(dòng)和 CUDA 工具包。從常見(jiàn)部署經(jīng)驗(yàn)看YOLOv8n 在 CPU 上處理單張 640x640 圖片通常需要幾百毫秒到 1 秒左右GPU 上可以到幾十毫秒甚至更快。顯存方面輕量模型在推理時(shí)占用量不大但實(shí)際數(shù)字需要結(jié)合模型版本、批量大小和輸入分辨率測(cè)試確認(rèn)。3.2 軟件環(huán)境建議使用 Python 3.8 到 3.11 之間的版本PyQt5 和 Ultralytics 在這幾個(gè)版本下兼容性比較好。核心依賴包括ultralytics PyQt5 opencv-python torch torchvision numpy Pillow其中torch和torchvision的安裝方式取決于是否使用 GPU。CPU 版本可以直接安裝默認(rèn)的 PyPI 包GPU 版本需要到 PyTorch 官網(wǎng)根據(jù) CUDA 版本選擇對(duì)應(yīng)安裝命令。3.3 環(huán)境檢查依賴安裝完成后建議先跑一個(gè)簡(jiǎn)單的 YOLOv8 推理驗(yàn)證環(huán)境是否正常from ultralytics import YOLO model YOLO(yolov8n.pt) results model(test.jpg) results[0].show()如果界面彈出了檢測(cè)結(jié)果窗口說(shuō)明核心依賴已經(jīng)可用。接下來(lái)就可以進(jìn)入項(xiàng)目代碼編寫(xiě)階段。4. YOLOv8 麥穗稻穗檢測(cè)模型準(zhǔn)備4.1 數(shù)據(jù)集標(biāo)注與目錄結(jié)構(gòu)訓(xùn)練自定義檢測(cè)模型的第一步是準(zhǔn)備數(shù)據(jù)集。麥穗稻穗檢測(cè)屬于單類別或雙類別目標(biāo)檢測(cè)任務(wù)建議每類目標(biāo)準(zhǔn)備 1000 張以上的標(biāo)注圖片。圖片可以從田間多角度、多光照條件拍攝也可以從公開(kāi)農(nóng)業(yè)數(shù)據(jù)集補(bǔ)充。推薦使用 LabelImg 或 X-AnyLabeling 進(jìn)行標(biāo)注標(biāo)注格式選擇 YOLO 格式即每張圖片對(duì)應(yīng)一個(gè)同名的.txt文件文件中每行內(nèi)容為類別id x_center y_center width height其中坐標(biāo)值均為歸一化后的數(shù)值。數(shù)據(jù)集目錄結(jié)構(gòu)如下dataset/ ├── images/ │ ├── train/ │ └── val/ ├── labels/ │ ├── train/ │ └── val/ └── data.yamldata.yaml內(nèi)容示例path: dataset train: images/train val: images/val nc: 2 names: [wheat_head, rice_head]4.2 模型訓(xùn)練在項(xiàng)目根目錄執(zhí)行訓(xùn)練命令。以 YOLOv8n 為例yolo detect train datadata.yaml modelyolov8n.pt epochs100 imgsz640 batch16batch 大小需要根據(jù)顯存調(diào)整。顯存較小的設(shè)備可以設(shè)置batch8或batch4同時(shí)開(kāi)啟cacheTrue可以加快數(shù)據(jù)加載。訓(xùn)練完成后最佳權(quán)重保存在runs/detect/train/weights/best.pt。如果是 GTX 1660 Ti 這類 6G 顯存的顯卡YOLOv8n 或者 YOLOv8s 配合 640 輸入分辨率是可以正常訓(xùn)練的YOLOv8m 以上需要把 batch 調(diào)小或者降低輸入尺寸。如果訓(xùn)練時(shí)遇到CUDA out of memory優(yōu)先減 batch 而不是降分辨率因?yàn)榉直媛蕦?duì)模型精度影響更大。4.3 推理測(cè)試訓(xùn)練完成后可以用 best.pt 做一次推理測(cè)試from ultralytics import YOLO model YOLO(runs/detect/train/weights/best.pt) results model(test_image.jpg, conf0.25) results[0].show()確認(rèn)模型在驗(yàn)證集圖片上能夠正確框出麥穗稻穗保存該模型文件路徑后面會(huì)用到 PyQt5 界面中。5. PyQt5 界面整體設(shè)計(jì)5.1 功能模塊拆分PyQt5 界面部分建議拆成幾個(gè)獨(dú)立的模塊避免把所有邏輯都堆在main.py里主窗口模塊負(fù)責(zé)整體布局、菜單欄、按鈕事件綁定推理線程模塊負(fù)責(zé)調(diào)用 YOLOv8 模型進(jìn)行檢測(cè)避免阻塞 UI繪圖模塊負(fù)責(zé)在 QLabel 或 QGraphicsView 上繪制檢測(cè)框工具模塊負(fù)責(zé)文件選擇、結(jié)果保存、數(shù)量統(tǒng)計(jì)5.2 主界面布局推薦使用左側(cè)控制區(qū) 右側(cè)顯示區(qū)的布局結(jié)構(gòu)。左側(cè)放置圖片/視頻/攝像頭切換按鈕、置信度閾值滑塊、檢測(cè)按鈕和結(jié)果統(tǒng)計(jì)表格右側(cè)使用 QStackedWidget 切換靜態(tài)圖片顯示和視頻流顯示。from PyQt5.QtWidgets import ( QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QSlider, QTableWidget, QFileDialog ) class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle(麥穗稻穗檢測(cè)識(shí)別系統(tǒng) - YOLOv8 PyQt5) self.resize(1200, 800) self.init_ui() def init_ui(self): central_widget QWidget() self.setCentralWidget(central_widget) main_layout QHBoxLayout(central_widget) # 左側(cè)控制面板 control_panel QWidget() control_layout QVBoxLayout(control_panel) self.btn_image QPushButton(選擇圖片) self.btn_video QPushButton(選擇視頻) self.btn_camera QPushButton(打開(kāi)攝像頭) self.btn_detect QPushButton(執(zhí)行檢測(cè)) self.btn_save QPushButton(保存結(jié)果) control_layout.addWidget(self.btn_image) control_layout.addWidget(self.btn_video) control_layout.addWidget(self.btn_camera) control_layout.addWidget(self.btn_detect) control_layout.addWidget(self.btn_save) # 右側(cè)顯示區(qū)域 display_panel QWidget() display_layout QVBoxLayout(display_panel) self.image_label QLabel(圖像顯示區(qū)域) self.image_label.setMinimumSize(800, 600) display_layout.addWidget(self.image_label) main_layout.addWidget(control_panel) main_layout.addWidget(display_panel)5.3 推理線程實(shí)現(xiàn)PyQt5 的界面線程必須保持響應(yīng)耗時(shí)的目標(biāo)檢測(cè)不能直接寫(xiě)在按鈕的槽函數(shù)里否則界面會(huì)卡死。推薦使用QThread或QRunnable QThreadPool來(lái)做異步推理。import cv2 import numpy as np from PyQt5.QtCore import QThread, pyqtSignal from ultralytics import YOLO class DetectThread(QThread): result_ready pyqtSignal(np.ndarray, list, int) def __init__(self, model_path, image_path, conf_thres0.25): super().__init__() self.model YOLO(model_path) self.image_path image_path self.conf_thres conf_thres def run(self): image cv2.imread(self.image_path) if image is None: return results self.model(image, confself.conf_thres)[0] boxes results.boxes.xyxy.cpu().numpy() classes results.boxes.cls.cpu().numpy() confs results.boxes.conf.cpu().numpy() names results.names detections [] for box, cls, conf in zip(boxes, classes, confs): detections.append({ bbox: [int(v) for v in box], class: names[int(cls)], confidence: float(conf) }) annotated_frame results.plot() annotated_frame cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB) self.result_ready.emit(annotated_frame, detections, len(detections))關(guān)鍵點(diǎn)是results.plot()已經(jīng)幫我們把檢測(cè)框、類別標(biāo)簽和置信度畫(huà)到了圖像上省去了手動(dòng)用 OpenCV 繪制的步驟。如果希望統(tǒng)一界面風(fēng)格也可以自己遍歷 detections 并用cv2.rectangle繪制。5.4 主窗口與線程交互選擇圖片后將該路徑傳給 DetectThread并在result_ready信號(hào)中接收檢測(cè)結(jié)果def on_detect_clicked(self): if not hasattr(self, current_image_path): return self.detect_thread DetectThread( model_pathruns/detect/train/weights/best.pt, image_pathself.current_image_path, conf_thresself.conf_slider.value() / 100.0 ) self.detect_thread.result_ready.connect(self.on_result) self.detect_thread.start() def on_result(self, annotated_frame, detections, count): height, width, channel annotated_frame.shape bytes_per_line channel * width q_image QImage(annotated_frame.data, width, height, bytes_per_line, QImage.Format_RGB888) pixmap QPixmap.fromImage(q_image) self.image_label.setPixmap(pixmap.scaled( self.image_label.size(), Qt.KeepAspectRatio)) self.update_table(detections) self.label_count.setText(f目標(biāo)數(shù)量: {count})通過(guò) QThread 做異步推理后點(diǎn)擊檢測(cè)按鈕會(huì)立即觸發(fā)線程界面不卡頓。圖片較大時(shí)QImage 轉(zhuǎn)換后要記得調(diào)用setPixmap前先釋放之前的 pixmap 引用避免內(nèi)存增長(zhǎng)。6. 三種輸入模式與實(shí)時(shí)檢測(cè)實(shí)現(xiàn)6.1 圖片檢測(cè)模式圖片檢測(cè)是最簡(jiǎn)單也最穩(wěn)定的模式直接讀取本地圖片文件推理結(jié)束后把結(jié)果展示在界面上。建議在檢測(cè)結(jié)果下方同時(shí)展示原始圖片和標(biāo)注圖片的切換按鈕方便對(duì)比。6.2 視頻文件檢測(cè)模式視頻檢測(cè)的處理邏輯與圖片不同需要逐幀讀取、逐幀檢測(cè)。為了保持流暢度不需要每幀都送入模型可以設(shè)置跳幀策略比如每 2 幀檢測(cè)一次。視頻檢測(cè)放到獨(dú)立的 QThread 中通過(guò) OpenCV 的VideoCapture讀取視頻幀推理后將結(jié)果轉(zhuǎn)換為 QImage 發(fā)送到界面線程。class VideoDetectThread(QThread): frame_ready pyqtSignal(np.ndarray) def __init__(self, model_path, video_path, conf_thres0.25): super().__init__() self.model YOLO(model_path) self.video_path video_path self.conf_thres conf_thres self.is_running True def run(self): cap cv2.VideoCapture(self.video_path) frame_count 0 while self.is_running and cap.isOpened(): ret, frame cap.read() if not ret: break frame_count 1 if frame_count % 2 ! 0: continue results self.model(frame, confself.conf_thres)[0] annotated_frame results.plot() self.frame_ready.emit(annotated_frame) cap.release()視頻檢測(cè)中的核心問(wèn)題是線程退出機(jī)制必須在窗口關(guān)閉事件中設(shè)置is_running False并調(diào)用wait()等待線程結(jié)束否則會(huì)出現(xiàn)程序無(wú)法退出或崩潰的問(wèn)題。6.3 攝像頭實(shí)時(shí)檢測(cè)模式攝像頭模式實(shí)現(xiàn)邏輯與視頻文件相似只需要把VideoCapture的參數(shù)從文件路徑改為攝像頭索引cap cv2.VideoCapture(0) # 0 表示默認(rèn)攝像頭攝像頭采集的幀率一般是 30fps 左右但 YOLOv8 的推理速度通常達(dá)不到每秒 30 幀因此必須做跳幀處理。另外在低成本設(shè)備上建議把攝像頭輸入分辨率調(diào)低例cap.set(cv2.CAP_PROP_FRAME_WIDTH, 960) cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 540)降低輸入分辨率可以顯著減少 YOLOv8 的推理耗時(shí)。做農(nóng)忙季節(jié)田間移動(dòng)端部署時(shí)這比換大模型更實(shí)用。7. 檢測(cè)結(jié)果統(tǒng)計(jì)與保存7.1 結(jié)果統(tǒng)計(jì)表PyQt5 中可以使用QTableWidget展示檢測(cè)結(jié)果的詳細(xì)數(shù)據(jù)def update_table(self, detections): self.table.setRowCount(len(detections)) for row, det in enumerate(detections): bbox det[bbox] self.table.setItem(row, 0, QTableWidgetItem(det[class])) self.table.setItem(row, 1, QTableWidgetItem(f{det[confidence]:.2f})) self.table.setItem(row, 2, QTableWidgetItem(f({bbox[0]}, {bbox[1]}))) self.table.setItem(row, 3, QTableWidgetItem(f({bbox[2]}, {bbox[3]})))結(jié)果表可以實(shí)時(shí)記錄每個(gè)檢測(cè)框的坐標(biāo)和置信度。對(duì)于麥穗計(jì)數(shù)場(chǎng)景直接看表格中的行數(shù)即可。如果希望進(jìn)一步做統(tǒng)計(jì)分析可以將表格數(shù)據(jù)導(dǎo)出為 CSVdef save_results(self): path, _ QFileDialog.getSaveFileName( self, 保存檢測(cè)結(jié)果, , CSV Files (*.csv)) if not path: return with open(path, w, encodingutf-8) as f: f.write(class,confidence,x1,y1,x2,y2\n) for row in range(self.table.rowCount()): items [ self.table.item(row, col).text() for col in range(self.table.columnCount()) ] f.write(,.join(items) \n)CSV 導(dǎo)出是農(nóng)業(yè)科研場(chǎng)景里很實(shí)用的功能后續(xù)可以直接導(dǎo)入 Excel 或 SPSS 做產(chǎn)量統(tǒng)計(jì)、密度分析。8. 性能優(yōu)化與資源占用觀察8.1 CPU vs GPU 推理差異YOLOv8 在 CPU 上做推理完全可行但速度受模型尺寸影響很大。YOLOv8n 在 CPU 上速度尚可YOLOv8s 開(kāi)始就比較吃力了YOLOv8m 及以上基本建議走 GPU。如果只有 CPU 平臺(tái)可以嘗試將模型導(dǎo)出為 OpenVINO 格式來(lái)加速yolo export modelbest.pt formatopenvinoOpenVINO 格式的模型在英特爾 CPU 上通常能獲得 1.5 到 3 倍的加速效果。8.2 顯存占用觀察方法訓(xùn)練時(shí)用nvidia-smi命令實(shí)時(shí)查看顯存占用nvidia-smi -l 1推理階段顯存占用通常遠(yuǎn)低于訓(xùn)練階段。常見(jiàn) 6G 顯卡在 YOLOv8n 推理時(shí)可能只占 1G 到 2G 左右但這因輸入分辨率和批次大小而異具體數(shù)字需要在本機(jī)實(shí)測(cè)。8.3 降低資源占用的有效手段輸入分辨率從 1280 降到 640顯存和耗時(shí)都會(huì)明顯下降使用halfTrue開(kāi)啟 FP16 推理減小批量大小使用模型導(dǎo)出后的 int8 量化版本關(guān)閉不用的后臺(tái)進(jìn)程視頻檢測(cè)跳幀策略也是降低整體資源消耗的有效手段田間監(jiān)控視頻場(chǎng)景下每 5 幀檢測(cè)一次也可以接受。9. 完整項(xiàng)目結(jié)構(gòu)參考wheat_rice_detector/ ├── main.py # 程序入口 ├── ui/ │ ├── main_window.py # 主窗口 │ └── styles.py # 樣式表 ├── core/ │ ├── detector.py # YOLOv8 封裝 │ └── threads.py # 推理線程 ├── models/ │ └── best.pt # 訓(xùn)練好的模型權(quán)重 ├── dataset/ │ ├── images/ │ ├── labels/ │ └── data.yaml ├── outputs/ # 檢測(cè)結(jié)果輸出目錄 └── requirements.txt # 依賴列表detector.py中把 YOLOv8 的加載和推理邏輯單獨(dú)封裝from ultralytics import YOLO class Detector: def __init__(self, model_pathyolov8n.pt): self.model YOLO(model_path) def predict(self, image, conf_thres0.25): results self.model(image, confconf_thres)[0] return results這樣主界面代碼只需要?jiǎng)?chuàng)建 Detector 實(shí)例并調(diào)用predict方法不需要關(guān)心 YOLO 內(nèi)部細(xì)節(jié)。線程、界面和模型邏輯解耦后后續(xù)如果要接入新的檢測(cè)算法如 YOLOv9、YOLO11只需要修改Detector類的內(nèi)部實(shí)現(xiàn)即可。10. PyInstaller 打包發(fā)布如果要把系統(tǒng)交付給非開(kāi)發(fā)人員需要打包成 exe 文件。PyInstaller 打包 PyQt5 Ultralytics 項(xiàng)目有幾個(gè)注意事項(xiàng)。10.1 安裝 PyInstallerpip install pyinstaller10.2 生成 spec 文件PyInstaller 打包時(shí)需要把模型文件和資源文件作為附加數(shù)據(jù)加入pyinstaller --name WheatRiceDetector --onefile --windowed ^ --add-data models/best.pt;models ^ --hidden-import ultralytics ^ main.py如果是在 Linux 或 macOS 環(huán)境路徑分隔符要換成冒號(hào)pyinstaller --name WheatRiceDetector --onefile --windowed \ --add-data models/best.pt:models \ --hidden-import ultralytics \ main.py10.3 打包后常見(jiàn)問(wèn)題使用--onefile模式打包的 exe 啟動(dòng)時(shí)會(huì)先解壓到臨時(shí)目錄啟動(dòng)速度會(huì)變慢而且容易被殺毒軟件誤報(bào)。如果對(duì)體積和啟動(dòng)速度有要求可以使用--onedir模式也能避免一些誤報(bào)問(wèn)題。--windowed會(huì)隱藏控制臺(tái)窗口但檢測(cè)過(guò)程中的 print 信息也就看不到了。如果需要在程序中打印日志建議用 logging 同時(shí)輸出到文件和彈窗不要依賴控制臺(tái)。打包后 exe 運(yùn)行在另一臺(tái)電腦上如果對(duì)方電腦沒(méi)有顯卡驅(qū)動(dòng)或者沒(méi)有安裝 CUDAtorch 在導(dǎo)入時(shí)可能報(bào)錯(cuò)。穩(wěn)妥的方案是打包純 CPU 版本的 torch雖然推理速度慢一些但兼容性最好。11. 常見(jiàn)問(wèn)題與排查方法問(wèn)題現(xiàn)象可能原因排查方式解決方案界面啟動(dòng)后立即閃退PyQt5 依賴缺失在 IDE 中運(yùn)行 main.py 查看報(bào)錯(cuò)安裝 PyQt5 及對(duì)應(yīng)缺少的依賴模型文件加載失敗best.pt 路徑錯(cuò)誤或文件損壞檢查路徑是否存在嘗試重新加載使用絕對(duì)路徑或修正路徑重新訓(xùn)練/下載權(quán)重檢測(cè)按鈕點(diǎn)擊無(wú)反應(yīng)線程內(nèi)異?；蚰Ｐ屯评沓鲥e(cuò)在 DetectThread.run 中加 try-except 打印異常檢查圖像是否為中文路徑確認(rèn)模型與圖像尺寸是否兼容CUDA out of memory顯存不足查看nvidia-smi確認(rèn)顯存狀態(tài)減小 batch、降低 imgsz、換輕量模型視頻檢測(cè)畫(huà)面卡頓推理耗時(shí)超過(guò)幀間隔打印每幀推理耗時(shí)增加跳幀數(shù)、降低輸入分辨率、使用更小的模型攝像頭檢測(cè)黑屏攝像頭索引錯(cuò)誤或設(shè)備被占用換cv2.VideoCapture(1)測(cè)試關(guān)閉占用攝像頭的其他程序檢查設(shè)備管理器打包后 exe 缺少模型--add-data 未正確配置檢查打包時(shí)配置重新打包確保模型路徑已加入 data 目錄PyQt5 下拉框閃退信號(hào)連接方式或?qū)ο笊芷趩?wèn)題檢查槽函數(shù)中對(duì)象引用為控件設(shè)置父對(duì)象避免 Python 垃圾回收提前釋放對(duì)象中文圖片路徑下視頻讀不出來(lái)OpenCV 無(wú)法直接讀取中文路徑用os.path.exists檢查路徑使用cv2.imdecode或先重命名文件為英文路徑用 QThread 做目標(biāo)檢測(cè)時(shí)一個(gè)非常隱蔽的坑是在按鈕的槽函數(shù)中創(chuàng)建了線程對(duì)象但槽函數(shù)執(zhí)行結(jié)束后線程對(duì)象被 Python 垃圾回收結(jié)果就是線程還沒(méi)結(jié)束對(duì)象先沒(méi)了信號(hào)發(fā)不出來(lái)。解決方法是把線程對(duì)象保存為 self 的成員變量并在窗口關(guān)閉事件中主動(dòng)等線程結(jié)束。12. 最佳實(shí)踐與工程化建議12.1 從訓(xùn)練到部署的完整鏈路先小規(guī)模訓(xùn)練驗(yàn)證數(shù)據(jù)標(biāo)注質(zhì)量跑通流程后再擴(kuò)數(shù)據(jù)集。第一次訓(xùn)練建議只跑 50 輪觀察 loss 曲線是否下降如果 loss 不降優(yōu)先檢查數(shù)據(jù)標(biāo)注是否有大量錯(cuò)標(biāo)漏標(biāo)。12.2 模型選擇策略麥穗稻穗目標(biāo)在田間圖片中往往密集排列且存在遮擋這屬于小目標(biāo)檢測(cè)場(chǎng)景。如果 YOLOv8n 精度不夠不要急著上 YOLOv8x先嘗試以下手段提高訓(xùn)練輸入分辨率到 960開(kāi)啟數(shù)據(jù)增強(qiáng)參數(shù)如hsv_h、hsv_s、degrees使用 YOLOv8m 并用更大的 batch 訓(xùn)練在頸部引入小目標(biāo)檢測(cè)頭嘗試 YOLO11 或 YOLOv5u 等替代模型12.3 目錄與命名規(guī)范數(shù)據(jù)集、模型權(quán)重、輸出結(jié)果分開(kāi)存放。輸出目錄建議按日期建子目錄避免歷史檢測(cè)結(jié)果被覆蓋。模型的訓(xùn)練權(quán)重和推理權(quán)重分開(kāi)保存防止誤用中途保存的 checkpoint。outputs/ ├── 20250101/ │ ├── image1_annotated.jpg │ ├── image1_results.csv │ └── video1_detected.mp4 └── 20250102/12.4 線程安全與異常處理推理線程內(nèi)部涉及模型調(diào)用、OpenCV 圖像處理、numpy 數(shù)組操作這些代碼在異常時(shí)會(huì)導(dǎo)致線程靜默退出。建議在線程運(yùn)行入口處加全局異常捕獲。def run(self): try: # 省略原有邏輯 pass except Exception as e: self.error_occurred.emit(str(e))界面端在收到錯(cuò)誤信號(hào)后彈窗提示用戶而不是等線程無(wú)聲消亡后讓用戶面對(duì)一個(gè)“無(wú)響應(yīng)”的按鈕。13. 總結(jié)與下一步這個(gè)項(xiàng)目的核心鏈路其實(shí)很清晰YOLOv8 提供檢測(cè)能力PyQt5 包裝出桌面界面兩者獨(dú)立維護(hù)、通過(guò)信號(hào)槽機(jī)制通信。相比單獨(dú)跑 YOLOv8 命令行推理這套系統(tǒng)讓非技術(shù)人員也能上手操作這是它最大的價(jià)值。如果你正在做類似的系統(tǒng)建議先復(fù)現(xiàn)圖片檢測(cè)這一條鏈路跑通后再加入視頻和攝像頭模式。訓(xùn)練模型用官方y(tǒng)olo命令即可重點(diǎn)精力花在 PyQt5 界面和線程設(shè)計(jì)上。最容易踩的兩個(gè)坑一個(gè)是 QThread 對(duì)象的生命周期一個(gè)是打包時(shí)模型文件的掛載這兩點(diǎn)解決掉系統(tǒng)基本就能穩(wěn)定運(yùn)行。下一步可以考慮的方向包括接入 RTSP 網(wǎng)絡(luò)攝像頭實(shí)現(xiàn)田間遠(yuǎn)程監(jiān)測(cè)把檢測(cè)結(jié)果同步到 Excel 或數(shù)據(jù)庫(kù)在界面中加入數(shù)據(jù)分布直方圖實(shí)時(shí)展示不同地塊的麥穗密度對(duì)比或者把模型封裝成本地 API 服務(wù)通過(guò) HTTP 接口對(duì)接到 Web 端或小程序端。架構(gòu)層面這套 YOLOv8 PyQt5 的組合并不局限于麥穗稻穗檢測(cè)替換訓(xùn)練數(shù)據(jù)集后完全可以擴(kuò)展到其他農(nóng)作物計(jì)數(shù)、果園果實(shí)檢測(cè)或者工業(yè)零件識(shí)別場(chǎng)景。 基于YOLOv8和PyQt5的麥穗稻穗檢測(cè)識(shí)別系統(tǒng)設(shè)計(jì)與實(shí)現(xiàn)
這次我們來(lái)看一個(gè)把目標(biāo)檢測(cè)算法和桌面端工具結(jié)合得很典型的項(xiàng)目基于 YOLOv8 PyQt5 的麥穗稻穗檢測(cè)識(shí)別系統(tǒng)。這個(gè)項(xiàng)目本身不是新概念但它的價(jià)值在于落地形態(tài)很完整。YOLOv8 負(fù)責(zé)核心的麥穗稻穗目標(biāo)檢測(cè)PyQt5 負(fù)責(zé)提供可視化的桌面交互界面二者通過(guò) Python 生態(tài)整合最終產(chǎn)出的是一個(gè)可以獨(dú)立運(yùn)行的桌面應(yīng)用程序。相比單純跑一段檢測(cè)腳本這種帶圖形界面的系統(tǒng)更適合農(nóng)業(yè)科研場(chǎng)景中的樣本統(tǒng)計(jì)、田間調(diào)查記錄也非常適合作為畢業(yè)設(shè)計(jì)、課程項(xiàng)目或者工業(yè)視覺(jué)的參考原型。這篇文章會(huì)重點(diǎn)覆蓋這幾個(gè)部分YOLOv8 模型如何準(zhǔn)備和訓(xùn)練、PyQt5 桌面界面如何搭建、檢測(cè)結(jié)果如何在界面上實(shí)時(shí)展示、攝像頭/圖片/視頻三種輸入模式怎么切換、界面與推理線程怎么解耦避免卡頓以及最終如何打包成可分發(fā)的一鍵運(yùn)行程序。如果你正在做類似的檢測(cè)識(shí)別系統(tǒng)這篇文章可以直接參考整體架構(gòu)和關(guān)鍵代碼。1. 核心能力速覽能力項(xiàng)說(shuō)明項(xiàng)目類型桌面端目標(biāo)檢測(cè)識(shí)別系統(tǒng)檢測(cè)算法YOLOv8Ultralytics YOLOv8界面框架PyQt5Python 5檢測(cè)目標(biāo)麥穗、稻穗可通過(guò)數(shù)據(jù)集擴(kuò)展輸入方式本地圖片、視頻文件、攝像頭實(shí)時(shí)畫(huà)面輸出內(nèi)容檢測(cè)框、類別標(biāo)簽、置信度、目標(biāo)數(shù)量統(tǒng)計(jì)啟動(dòng)方式Python 腳本啟動(dòng) / PyInstaller 打包為 exeGPU 支持支持 CUDA 加速也支持 CPU 推理顯存需求需按模型版本和輸入分辨率測(cè)試YOLOv8n 在 CPU 上也能運(yùn)行API 接口非本項(xiàng)目核心但 YOLOv8 檢測(cè)邏輯可獨(dú)立封裝批量任務(wù)可擴(kuò)展界面邏輯和推理邏輯分離后可接入批量目錄處理適合場(chǎng)景農(nóng)業(yè)科研統(tǒng)計(jì)、田間調(diào)查、教學(xué)演示、畢業(yè)設(shè)計(jì)、算法原型驗(yàn)證從項(xiàng)目定位來(lái)看這個(gè)系統(tǒng)并不追求在算法層面做出創(chuàng)新而是把 YOLOv8 的檢測(cè)能力和 PyQt5 的桌面交互能力打通。它解決的核心問(wèn)題是讓沒(méi)有命令行操作習(xí)慣的使用者也能通過(guò)鼠標(biāo)點(diǎn)擊完成目標(biāo)檢測(cè)和結(jié)果統(tǒng)計(jì)。2. 適用場(chǎng)景與使用邊界2.1 適合誰(shuí)用農(nóng)業(yè)科研人員、植保相關(guān)專業(yè)的學(xué)生、需要做作物生長(zhǎng)情況統(tǒng)計(jì)的研究團(tuán)隊(duì)是這個(gè)系統(tǒng)最直接的受眾。比如在田間拍攝一批麥穗圖片過(guò)去靠人工數(shù)穗數(shù)費(fèi)時(shí)費(fèi)力現(xiàn)在可以通過(guò)模型自動(dòng)標(biāo)注每個(gè)麥穗的位置并輸出數(shù)量統(tǒng)計(jì)。對(duì)于計(jì)算機(jī)視覺(jué)方向的學(xué)生來(lái)說(shuō)這個(gè)項(xiàng)目的工程價(jià)值更明顯。它演示了如何把 YOLOv8 模型封裝成可交互的桌面應(yīng)用這套架構(gòu)可以遷移到其他檢測(cè)任務(wù)中比如果實(shí)計(jì)數(shù)、工廠零件檢測(cè)、安防目標(biāo)識(shí)別等。2.2 能解決什么問(wèn)題檢測(cè)識(shí)別系統(tǒng)要解決的并不是“識(shí)別得有多準(zhǔn)”這一個(gè)問(wèn)題而是“從拿到圖像到得到結(jié)果”這個(gè)完整鏈路的問(wèn)題。具體包括圖片預(yù)覽和檢測(cè)結(jié)果疊加顯示檢測(cè)框坐標(biāo)、類別、置信度的讀取和保存單張圖片中目標(biāo)數(shù)量的自動(dòng)統(tǒng)計(jì)視頻流和攝像頭畫(huà)面中的連續(xù)目標(biāo)檢測(cè)檢測(cè)結(jié)果的導(dǎo)出方便后續(xù)數(shù)據(jù)分析2.3 不適合什么場(chǎng)景實(shí)時(shí)性要求特別高的場(chǎng)景不適合直接用 PyQt5 的簡(jiǎn)單方案界面渲染本身會(huì)占用一定資源嵌入式設(shè)備部署也不是這個(gè)項(xiàng)目的強(qiáng)項(xiàng)更適合用 YOLOv8 導(dǎo)出 ONNX 或 TensorRT 后在 RK3588、Jetson 等邊緣設(shè)備上單獨(dú)部署大規(guī)模并行推理服務(wù)場(chǎng)景則建議把檢測(cè)邏輯獨(dú)立成 API 服務(wù)而不是依賴桌面界面。2.4 版權(quán)與授權(quán)邊界使用 YOLOv8 開(kāi)源框架時(shí)需要注意 Ultralytics 的許可證約定。訓(xùn)練數(shù)據(jù)如果是自己采集的田間圖片需要確保拍攝對(duì)象不涉及他人隱私或保密區(qū)域。如果使用公開(kāi)數(shù)據(jù)集則要遵循數(shù)據(jù)集的授權(quán)協(xié)議。系統(tǒng)如果涉及商業(yè)交付需要確認(rèn)模型權(quán)重、數(shù)據(jù)集和代碼部分的許可兼容性。3. 環(huán)境準(zhǔn)備與前置條件3.1 硬件環(huán)境YOLOv8 官方提供多種尺寸的模型n、s、m、l、x。其中 YOLOv8n 最輕量CPU 也能推理YOLOv8s 和 YOLOv8m 在普通消費(fèi)級(jí)顯卡上有比較好的精度和速度平衡。如果手頭只有核顯或者無(wú)獨(dú)顯設(shè)備優(yōu)先選 YOLOv8n。如果使用 NVIDIA 顯卡并希望通過(guò) CUDA 加速需要提前裝好顯卡驅(qū)動(dòng)和 CUDA 工具包。從常見(jiàn)部署經(jīng)驗(yàn)看YOLOv8n 在 CPU 上處理單張 640x640 圖片通常需要幾百毫秒到 1 秒左右GPU 上可以到幾十毫秒甚至更快。顯存方面輕量模型在推理時(shí)占用量不大但實(shí)際數(shù)字需要結(jié)合模型版本、批量大小和輸入分辨率測(cè)試確認(rèn)。3.2 軟件環(huán)境建議使用 Python 3.8 到 3.11 之間的版本PyQt5 和 Ultralytics 在這幾個(gè)版本下兼容性比較好。核心依賴包括ultralytics PyQt5 opencv-python torch torchvision numpy Pillow其中torch和torchvision的安裝方式取決于是否使用 GPU。CPU 版本可以直接安裝默認(rèn)的 PyPI 包GPU 版本需要到 PyTorch 官網(wǎng)根據(jù) CUDA 版本選擇對(duì)應(yīng)安裝命令。3.3 環(huán)境檢查依賴安裝完成后建議先跑一個(gè)簡(jiǎn)單的 YOLOv8 推理驗(yàn)證環(huán)境是否正常from ultralytics import YOLO model YOLO(yolov8n.pt) results model(test.jpg) results[0].show()如果界面彈出了檢測(cè)結(jié)果窗口說(shuō)明核心依賴已經(jīng)可用。接下來(lái)就可以進(jìn)入項(xiàng)目代碼編寫(xiě)階段。4. YOLOv8 麥穗稻穗檢測(cè)模型準(zhǔn)備4.1 數(shù)據(jù)集標(biāo)注與目錄結(jié)構(gòu)訓(xùn)練自定義檢測(cè)模型的第一步是準(zhǔn)備數(shù)據(jù)集。麥穗稻穗檢測(cè)屬于單類別或雙類別目標(biāo)檢測(cè)任務(wù)建議每類目標(biāo)準(zhǔn)備 1000 張以上的標(biāo)注圖片。圖片可以從田間多角度、多光照條件拍攝也可以從公開(kāi)農(nóng)業(yè)數(shù)據(jù)集補(bǔ)充。推薦使用 LabelImg 或 X-AnyLabeling 進(jìn)行標(biāo)注標(biāo)注格式選擇 YOLO 格式即每張圖片對(duì)應(yīng)一個(gè)同名的.txt文件文件中每行內(nèi)容為類別id x_center y_center width height其中坐標(biāo)值均為歸一化后的數(shù)值。數(shù)據(jù)集目錄結(jié)構(gòu)如下dataset/ ├── images/ │ ├── train/ │ └── val/ ├── labels/ │ ├── train/ │ └── val/ └── data.yamldata.yaml內(nèi)容示例path: dataset train: images/train val: images/val nc: 2 names: [wheat_head, rice_head]4.2 模型訓(xùn)練在項(xiàng)目根目錄執(zhí)行訓(xùn)練命令。以 YOLOv8n 為例yolo detect train datadata.yaml modelyolov8n.pt epochs100 imgsz640 batch16batch 大小需要根據(jù)顯存調(diào)整。顯存較小的設(shè)備可以設(shè)置batch8或batch4同時(shí)開(kāi)啟cacheTrue可以加快數(shù)據(jù)加載。訓(xùn)練完成后最佳權(quán)重保存在runs/detect/train/weights/best.pt。如果是 GTX 1660 Ti 這類 6G 顯存的顯卡YOLOv8n 或者 YOLOv8s 配合 640 輸入分辨率是可以正常訓(xùn)練的YOLOv8m 以上需要把 batch 調(diào)小或者降低輸入尺寸。如果訓(xùn)練時(shí)遇到CUDA out of memory優(yōu)先減 batch 而不是降分辨率因?yàn)榉直媛蕦?duì)模型精度影響更大。4.3 推理測(cè)試訓(xùn)練完成后可以用 best.pt 做一次推理測(cè)試from ultralytics import YOLO model YOLO(runs/detect/train/weights/best.pt) results model(test_image.jpg, conf0.25) results[0].show()確認(rèn)模型在驗(yàn)證集圖片上能夠正確框出麥穗稻穗保存該模型文件路徑后面會(huì)用到 PyQt5 界面中。5. PyQt5 界面整體設(shè)計(jì)5.1 功能模塊拆分PyQt5 界面部分建議拆成幾個(gè)獨(dú)立的模塊避免把所有邏輯都堆在main.py里主窗口模塊負(fù)責(zé)整體布局、菜單欄、按鈕事件綁定推理線程模塊負(fù)責(zé)調(diào)用 YOLOv8 模型進(jìn)行檢測(cè)避免阻塞 UI繪圖模塊負(fù)責(zé)在 QLabel 或 QGraphicsView 上繪制檢測(cè)框工具模塊負(fù)責(zé)文件選擇、結(jié)果保存、數(shù)量統(tǒng)計(jì)5.2 主界面布局推薦使用左側(cè)控制區(qū) 右側(cè)顯示區(qū)的布局結(jié)構(gòu)。左側(cè)放置圖片/視頻/攝像頭切換按鈕、置信度閾值滑塊、檢測(cè)按鈕和結(jié)果統(tǒng)計(jì)表格右側(cè)使用 QStackedWidget 切換靜態(tài)圖片顯示和視頻流顯示。from PyQt5.QtWidgets import ( QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QSlider, QTableWidget, QFileDialog ) class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle(麥穗稻穗檢測(cè)識(shí)別系統(tǒng) - YOLOv8 PyQt5) self.resize(1200, 800) self.init_ui() def init_ui(self): central_widget QWidget() self.setCentralWidget(central_widget) main_layout QHBoxLayout(central_widget) # 左側(cè)控制面板 control_panel QWidget() control_layout QVBoxLayout(control_panel) self.btn_image QPushButton(選擇圖片) self.btn_video QPushButton(選擇視頻) self.btn_camera QPushButton(打開(kāi)攝像頭) self.btn_detect QPushButton(執(zhí)行檢測(cè)) self.btn_save QPushButton(保存結(jié)果) control_layout.addWidget(self.btn_image) control_layout.addWidget(self.btn_video) control_layout.addWidget(self.btn_camera) control_layout.addWidget(self.btn_detect) control_layout.addWidget(self.btn_save) # 右側(cè)顯示區(qū)域 display_panel QWidget() display_layout QVBoxLayout(display_panel) self.image_label QLabel(圖像顯示區(qū)域) self.image_label.setMinimumSize(800, 600) display_layout.addWidget(self.image_label) main_layout.addWidget(control_panel) main_layout.addWidget(display_panel)5.3 推理線程實(shí)現(xiàn)PyQt5 的界面線程必須保持響應(yīng)耗時(shí)的目標(biāo)檢測(cè)不能直接寫(xiě)在按鈕的槽函數(shù)里否則界面會(huì)卡死。推薦使用QThread或QRunnable QThreadPool來(lái)做異步推理。import cv2 import numpy as np from PyQt5.QtCore import QThread, pyqtSignal from ultralytics import YOLO class DetectThread(QThread): result_ready pyqtSignal(np.ndarray, list, int) def __init__(self, model_path, image_path, conf_thres0.25): super().__init__() self.model YOLO(model_path) self.image_path image_path self.conf_thres conf_thres def run(self): image cv2.imread(self.image_path) if image is None: return results self.model(image, confself.conf_thres)[0] boxes results.boxes.xyxy.cpu().numpy() classes results.boxes.cls.cpu().numpy() confs results.boxes.conf.cpu().numpy() names results.names detections [] for box, cls, conf in zip(boxes, classes, confs): detections.append({ bbox: [int(v) for v in box], class: names[int(cls)], confidence: float(conf) }) annotated_frame results.plot() annotated_frame cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB) self.result_ready.emit(annotated_frame, detections, len(detections))關(guān)鍵點(diǎn)是results.plot()已經(jīng)幫我們把檢測(cè)框、類別標(biāo)簽和置信度畫(huà)到了圖像上省去了手動(dòng)用 OpenCV 繪制的步驟。如果希望統(tǒng)一界面風(fēng)格也可以自己遍歷 detections 并用cv2.rectangle繪制。5.4 主窗口與線程交互選擇圖片后將該路徑傳給 DetectThread并在result_ready信號(hào)中接收檢測(cè)結(jié)果def on_detect_clicked(self): if not hasattr(self, current_image_path): return self.detect_thread DetectThread( model_pathruns/detect/train/weights/best.pt, image_pathself.current_image_path, conf_thresself.conf_slider.value() / 100.0 ) self.detect_thread.result_ready.connect(self.on_result) self.detect_thread.start() def on_result(self, annotated_frame, detections, count): height, width, channel annotated_frame.shape bytes_per_line channel * width q_image QImage(annotated_frame.data, width, height, bytes_per_line, QImage.Format_RGB888) pixmap QPixmap.fromImage(q_image) self.image_label.setPixmap(pixmap.scaled( self.image_label.size(), Qt.KeepAspectRatio)) self.update_table(detections) self.label_count.setText(f目標(biāo)數(shù)量: {count})通過(guò) QThread 做異步推理后點(diǎn)擊檢測(cè)按鈕會(huì)立即觸發(fā)線程界面不卡頓。圖片較大時(shí)QImage 轉(zhuǎn)換后要記得調(diào)用setPixmap前先釋放之前的 pixmap 引用避免內(nèi)存增長(zhǎng)。6. 三種輸入模式與實(shí)時(shí)檢測(cè)實(shí)現(xiàn)6.1 圖片檢測(cè)模式圖片檢測(cè)是最簡(jiǎn)單也最穩(wěn)定的模式直接讀取本地圖片文件推理結(jié)束后把結(jié)果展示在界面上。建議在檢測(cè)結(jié)果下方同時(shí)展示原始圖片和標(biāo)注圖片的切換按鈕方便對(duì)比。6.2 視頻文件檢測(cè)模式視頻檢測(cè)的處理邏輯與圖片不同需要逐幀讀取、逐幀檢測(cè)。為了保持流暢度不需要每幀都送入模型可以設(shè)置跳幀策略比如每 2 幀檢測(cè)一次。視頻檢測(cè)放到獨(dú)立的 QThread 中通過(guò) OpenCV 的VideoCapture讀取視頻幀推理后將結(jié)果轉(zhuǎn)換為 QImage 發(fā)送到界面線程。class VideoDetectThread(QThread): frame_ready pyqtSignal(np.ndarray) def __init__(self, model_path, video_path, conf_thres0.25): super().__init__() self.model YOLO(model_path) self.video_path video_path self.conf_thres conf_thres self.is_running True def run(self): cap cv2.VideoCapture(self.video_path) frame_count 0 while self.is_running and cap.isOpened(): ret, frame cap.read() if not ret: break frame_count 1 if frame_count % 2 ! 0: continue results self.model(frame, confself.conf_thres)[0] annotated_frame results.plot() self.frame_ready.emit(annotated_frame) cap.release()視頻檢測(cè)中的核心問(wèn)題是線程退出機(jī)制必須在窗口關(guān)閉事件中設(shè)置is_running False并調(diào)用wait()等待線程結(jié)束否則會(huì)出現(xiàn)程序無(wú)法退出或崩潰的問(wèn)題。6.3 攝像頭實(shí)時(shí)檢測(cè)模式攝像頭模式實(shí)現(xiàn)邏輯與視頻文件相似只需要把VideoCapture的參數(shù)從文件路徑改為攝像頭索引cap cv2.VideoCapture(0) # 0 表示默認(rèn)攝像頭攝像頭采集的幀率一般是 30fps 左右但 YOLOv8 的推理速度通常達(dá)不到每秒 30 幀因此必須做跳幀處理。另外在低成本設(shè)備上建議把攝像頭輸入分辨率調(diào)低例cap.set(cv2.CAP_PROP_FRAME_WIDTH, 960) cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 540)降低輸入分辨率可以顯著減少 YOLOv8 的推理耗時(shí)。做農(nóng)忙季節(jié)田間移動(dòng)端部署時(shí)這比換大模型更實(shí)用。7. 檢測(cè)結(jié)果統(tǒng)計(jì)與保存7.1 結(jié)果統(tǒng)計(jì)表PyQt5 中可以使用QTableWidget展示檢測(cè)結(jié)果的詳細(xì)數(shù)據(jù)def update_table(self, detections): self.table.setRowCount(len(detections)) for row, det in enumerate(detections): bbox det[bbox] self.table.setItem(row, 0, QTableWidgetItem(det[class])) self.table.setItem(row, 1, QTableWidgetItem(f{det[confidence]:.2f})) self.table.setItem(row, 2, QTableWidgetItem(f({bbox[0]}, {bbox[1]}))) self.table.setItem(row, 3, QTableWidgetItem(f({bbox[2]}, {bbox[3]})))結(jié)果表可以實(shí)時(shí)記錄每個(gè)檢測(cè)框的坐標(biāo)和置信度。對(duì)于麥穗計(jì)數(shù)場(chǎng)景直接看表格中的行數(shù)即可。如果希望進(jìn)一步做統(tǒng)計(jì)分析可以將表格數(shù)據(jù)導(dǎo)出為 CSVdef save_results(self): path, _ QFileDialog.getSaveFileName( self, 保存檢測(cè)結(jié)果, , CSV Files (*.csv)) if not path: return with open(path, w, encodingutf-8) as f: f.write(class,confidence,x1,y1,x2,y2\n) for row in range(self.table.rowCount()): items [ self.table.item(row, col).text() for col in range(self.table.columnCount()) ] f.write(,.join(items) \n)CSV 導(dǎo)出是農(nóng)業(yè)科研場(chǎng)景里很實(shí)用的功能后續(xù)可以直接導(dǎo)入 Excel 或 SPSS 做產(chǎn)量統(tǒng)計(jì)、密度分析。8. 性能優(yōu)化與資源占用觀察8.1 CPU vs GPU 推理差異YOLOv8 在 CPU 上做推理完全可行但速度受模型尺寸影響很大。YOLOv8n 在 CPU 上速度尚可YOLOv8s 開(kāi)始就比較吃力了YOLOv8m 及以上基本建議走 GPU。如果只有 CPU 平臺(tái)可以嘗試將模型導(dǎo)出為 OpenVINO 格式來(lái)加速yolo export modelbest.pt formatopenvinoOpenVINO 格式的模型在英特爾 CPU 上通常能獲得 1.5 到 3 倍的加速效果。8.2 顯存占用觀察方法訓(xùn)練時(shí)用nvidia-smi命令實(shí)時(shí)查看顯存占用nvidia-smi -l 1推理階段顯存占用通常遠(yuǎn)低于訓(xùn)練階段。常見(jiàn) 6G 顯卡在 YOLOv8n 推理時(shí)可能只占 1G 到 2G 左右但這因輸入分辨率和批次大小而異具體數(shù)字需要在本機(jī)實(shí)測(cè)。8.3 降低資源占用的有效手段輸入分辨率從 1280 降到 640顯存和耗時(shí)都會(huì)明顯下降使用halfTrue開(kāi)啟 FP16 推理減小批量大小使用模型導(dǎo)出后的 int8 量化版本關(guān)閉不用的后臺(tái)進(jìn)程視頻檢測(cè)跳幀策略也是降低整體資源消耗的有效手段田間監(jiān)控視頻場(chǎng)景下每 5 幀檢測(cè)一次也可以接受。9. 完整項(xiàng)目結(jié)構(gòu)參考wheat_rice_detector/ ├── main.py # 程序入口 ├── ui/ │ ├── main_window.py # 主窗口 │ └── styles.py # 樣式表 ├── core/ │ ├── detector.py # YOLOv8 封裝 │ └── threads.py # 推理線程 ├── models/ │ └── best.pt # 訓(xùn)練好的模型權(quán)重 ├── dataset/ │ ├── images/ │ ├── labels/ │ └── data.yaml ├── outputs/ # 檢測(cè)結(jié)果輸出目錄 └── requirements.txt # 依賴列表detector.py中把 YOLOv8 的加載和推理邏輯單獨(dú)封裝from ultralytics import YOLO class Detector: def __init__(self, model_pathyolov8n.pt): self.model YOLO(model_path) def predict(self, image, conf_thres0.25): results self.model(image, confconf_thres)[0] return results這樣主界面代碼只需要?jiǎng)?chuàng)建 Detector 實(shí)例并調(diào)用predict方法不需要關(guān)心 YOLO 內(nèi)部細(xì)節(jié)。線程、界面和模型邏輯解耦后后續(xù)如果要接入新的檢測(cè)算法如 YOLOv9、YOLO11只需要修改Detector類的內(nèi)部實(shí)現(xiàn)即可。10. PyInstaller 打包發(fā)布如果要把系統(tǒng)交付給非開(kāi)發(fā)人員需要打包成 exe 文件。PyInstaller 打包 PyQt5 Ultralytics 項(xiàng)目有幾個(gè)注意事項(xiàng)。10.1 安裝 PyInstallerpip install pyinstaller10.2 生成 spec 文件PyInstaller 打包時(shí)需要把模型文件和資源文件作為附加數(shù)據(jù)加入pyinstaller --name WheatRiceDetector --onefile --windowed ^ --add-data models/best.pt;models ^ --hidden-import ultralytics ^ main.py如果是在 Linux 或 macOS 環(huán)境路徑分隔符要換成冒號(hào)pyinstaller --name WheatRiceDetector --onefile --windowed \ --add-data models/best.pt:models \ --hidden-import ultralytics \ main.py10.3 打包后常見(jiàn)問(wèn)題使用--onefile模式打包的 exe 啟動(dòng)時(shí)會(huì)先解壓到臨時(shí)目錄啟動(dòng)速度會(huì)變慢而且容易被殺毒軟件誤報(bào)。如果對(duì)體積和啟動(dòng)速度有要求可以使用--onedir模式也能避免一些誤報(bào)問(wèn)題。--windowed會(huì)隱藏控制臺(tái)窗口但檢測(cè)過(guò)程中的 print 信息也就看不到了。如果需要在程序中打印日志建議用 logging 同時(shí)輸出到文件和彈窗不要依賴控制臺(tái)。打包后 exe 運(yùn)行在另一臺(tái)電腦上如果對(duì)方電腦沒(méi)有顯卡驅(qū)動(dòng)或者沒(méi)有安裝 CUDAtorch 在導(dǎo)入時(shí)可能報(bào)錯(cuò)。穩(wěn)妥的方案是打包純 CPU 版本的 torch雖然推理速度慢一些但兼容性最好。11. 常見(jiàn)問(wèn)題與排查方法問(wèn)題現(xiàn)象可能原因排查方式解決方案界面啟動(dòng)后立即閃退PyQt5 依賴缺失在 IDE 中運(yùn)行 main.py 查看報(bào)錯(cuò)安裝 PyQt5 及對(duì)應(yīng)缺少的依賴模型文件加載失敗best.pt 路徑錯(cuò)誤或文件損壞檢查路徑是否存在嘗試重新加載使用絕對(duì)路徑或修正路徑重新訓(xùn)練/下載權(quán)重檢測(cè)按鈕點(diǎn)擊無(wú)反應(yīng)線程內(nèi)異?；蚰Ｐ屯评沓鲥e(cuò)在 DetectThread.run 中加 try-except 打印異常檢查圖像是否為中文路徑確認(rèn)模型與圖像尺寸是否兼容CUDA out of memory顯存不足查看nvidia-smi確認(rèn)顯存狀態(tài)減小 batch、降低 imgsz、換輕量模型視頻檢測(cè)畫(huà)面卡頓推理耗時(shí)超過(guò)幀間隔打印每幀推理耗時(shí)增加跳幀數(shù)、降低輸入分辨率、使用更小的模型攝像頭檢測(cè)黑屏攝像頭索引錯(cuò)誤或設(shè)備被占用換cv2.VideoCapture(1)測(cè)試關(guān)閉占用攝像頭的其他程序檢查設(shè)備管理器打包后 exe 缺少模型--add-data 未正確配置檢查打包時(shí)配置重新打包確保模型路徑已加入 data 目錄PyQt5 下拉框閃退信號(hào)連接方式或?qū)ο笊芷趩?wèn)題檢查槽函數(shù)中對(duì)象引用為控件設(shè)置父對(duì)象避免 Python 垃圾回收提前釋放對(duì)象中文圖片路徑下視頻讀不出來(lái)OpenCV 無(wú)法直接讀取中文路徑用os.path.exists檢查路徑使用cv2.imdecode或先重命名文件為英文路徑用 QThread 做目標(biāo)檢測(cè)時(shí)一個(gè)非常隱蔽的坑是在按鈕的槽函數(shù)中創(chuàng)建了線程對(duì)象但槽函數(shù)執(zhí)行結(jié)束后線程對(duì)象被 Python 垃圾回收結(jié)果就是線程還沒(méi)結(jié)束對(duì)象先沒(méi)了信號(hào)發(fā)不出來(lái)。解決方法是把線程對(duì)象保存為 self 的成員變量并在窗口關(guān)閉事件中主動(dòng)等線程結(jié)束。12. 最佳實(shí)踐與工程化建議12.1 從訓(xùn)練到部署的完整鏈路先小規(guī)模訓(xùn)練驗(yàn)證數(shù)據(jù)標(biāo)注質(zhì)量跑通流程后再擴(kuò)數(shù)據(jù)集。第一次訓(xùn)練建議只跑 50 輪觀察 loss 曲線是否下降如果 loss 不降優(yōu)先檢查數(shù)據(jù)標(biāo)注是否有大量錯(cuò)標(biāo)漏標(biāo)。12.2 模型選擇策略麥穗稻穗目標(biāo)在田間圖片中往往密集排列且存在遮擋這屬于小目標(biāo)檢測(cè)場(chǎng)景。如果 YOLOv8n 精度不夠不要急著上 YOLOv8x先嘗試以下手段提高訓(xùn)練輸入分辨率到 960開(kāi)啟數(shù)據(jù)增強(qiáng)參數(shù)如hsv_h、hsv_s、degrees使用 YOLOv8m 并用更大的 batch 訓(xùn)練在頸部引入小目標(biāo)檢測(cè)頭嘗試 YOLO11 或 YOLOv5u 等替代模型12.3 目錄與命名規(guī)范數(shù)據(jù)集、模型權(quán)重、輸出結(jié)果分開(kāi)存放。輸出目錄建議按日期建子目錄避免歷史檢測(cè)結(jié)果被覆蓋。模型的訓(xùn)練權(quán)重和推理權(quán)重分開(kāi)保存防止誤用中途保存的 checkpoint。outputs/ ├── 20250101/ │ ├── image1_annotated.jpg │ ├── image1_results.csv │ └── video1_detected.mp4 └── 20250102/12.4 線程安全與異常處理推理線程內(nèi)部涉及模型調(diào)用、OpenCV 圖像處理、numpy 數(shù)組操作這些代碼在異常時(shí)會(huì)導(dǎo)致線程靜默退出。建議在線程運(yùn)行入口處加全局異常捕獲。def run(self): try: # 省略原有邏輯 pass except Exception as e: self.error_occurred.emit(str(e))界面端在收到錯(cuò)誤信號(hào)后彈窗提示用戶而不是等線程無(wú)聲消亡后讓用戶面對(duì)一個(gè)“無(wú)響應(yīng)”的按鈕。13. 總結(jié)與下一步這個(gè)項(xiàng)目的核心鏈路其實(shí)很清晰YOLOv8 提供檢測(cè)能力PyQt5 包裝出桌面界面兩者獨(dú)立維護(hù)、通過(guò)信號(hào)槽機(jī)制通信。相比單獨(dú)跑 YOLOv8 命令行推理這套系統(tǒng)讓非技術(shù)人員也能上手操作這是它最大的價(jià)值。如果你正在做類似的系統(tǒng)建議先復(fù)現(xiàn)圖片檢測(cè)這一條鏈路跑通后再加入視頻和攝像頭模式。訓(xùn)練模型用官方y(tǒng)olo命令即可重點(diǎn)精力花在 PyQt5 界面和線程設(shè)計(jì)上。最容易踩的兩個(gè)坑一個(gè)是 QThread 對(duì)象的生命周期一個(gè)是打包時(shí)模型文件的掛載這兩點(diǎn)解決掉系統(tǒng)基本就能穩(wěn)定運(yùn)行。下一步可以考慮的方向包括接入 RTSP 網(wǎng)絡(luò)攝像頭實(shí)現(xiàn)田間遠(yuǎn)程監(jiān)測(cè)把檢測(cè)結(jié)果同步到 Excel 或數(shù)據(jù)庫(kù)在界面中加入數(shù)據(jù)分布直方圖實(shí)時(shí)展示不同地塊的麥穗密度對(duì)比或者把模型封裝成本地 API 服務(wù)通過(guò) HTTP 接口對(duì)接到 Web 端或小程序端。架構(gòu)層面這套 YOLOv8 PyQt5 的組合并不局限于麥穗稻穗檢測(cè)替換訓(xùn)練數(shù)據(jù)集后完全可以擴(kuò)展到其他農(nóng)作物計(jì)數(shù)、果園果實(shí)檢測(cè)或者工業(yè)零件識(shí)別場(chǎng)景。