计算机视觉 · YOLOv5

绿萝病害检测系统

基于 YOLOv5 的绿萝植物病害检测,从零构建数据集到模型部署的完整 CV 项目生命周期。

4
病害分类类别
400+
手工标注图片
<50ms
单图推理速度
8GB
显存约束适配

完整 CV 工作流

数据采集 LabelImg 标注 YOLOv5s 训练 ONNX 导出 GUI 部署

突破 A:从零构建数据集

绿萝病害无公开数据集可用——这是最大的起点障碍。整个数据工程过程:

突破 B:资源约束下的模型调优

硬件约束:RTX 4060 Laptop 仅 8GB 显存。在有限算力下完成全训练流程:

工程化部署形态

模型封装为本地 Python GUI 工具(tkinter 界面),支持两种使用模式:

覆盖了计算机视觉项目的完整生命周期:数据采集 → 标注 → 模型训练 → 导出优化 → 应用部署

技术栈

YOLOv5 (Ultralytics) PyTorch + CUDA 11.8 OpenCV DNN (ONNX) RTX 4060 Laptop LabelImg tkinter GUI
← 返回作品集首页