本次检测系统,不仅可以检测图片、视频或摄像头当中出现的草莓叶子是否有灰叶病,还可以检测出草莓叶是否缺钙、是否需要施肥等状态。基于最新的YOLO-v8训练的草莓生长状态检测模型和完整的python代码以及草莓的训练数据,下载后即可运行,输出检测结果。
文件夹说明
重要文件解释:predictWindow.py是Pyqt5界面展示主程序,并调用训练好的yolov8模型参数,进行生长状态类别的检测。stbleaftrain.py是训练脚本,detect_tools.py是用来读取和展示图像,被predictWindow.py调用.yolov8.pt是训练完成模型保存参数。
如有其它数据集检测需求,可定制项目
2.效果
需要代码可以关注
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog # 导入PyQt5库中的模块
import sys
import os
import glob
sys.path.append('UIProgram') # 添加一个路径到Python的模块搜索路径
from UIProgram.UiMain import Ui_MainWindow # 导入UI程序的主窗口定义
from UIProgram.QssLoader import QSSLoader # 导入自定义的QSSLoader模块
from PyQt5.QtCore import QTimer, Qt, QCoreApplication # 导入PyQt5库中的模块
import sys
from PyQt5.QtCore import Qt, QCoreApplication # 导入PyQt5库中的模块
from ultralytics import YOLO
import cv2
import detect_tools as tools
#代码和数据集:https://mbd.pub/o/bread/mbd-ZZ6Vl5xw