YOLO-v5实战应用：医疗废弃物识别分类系统
1. 引言
1.1 业务场景描述
随着城市化进程加快，医疗废弃物的产生量逐年上升。这类废弃物包括针头、纱布、废弃药品包装、一次性医疗器械等，若处理不当，极易造成环境污染和疾病传播。传统的医疗废弃物分类依赖人工操作，效率低且存在安全风险。因此，构建一个自动化、高精度的识别与分类系统成为医疗机构和环保单位的迫切需求。
计算机视觉技术的发展为这一问题提供了新的解决路径。通过在关键区域部署摄像头并结合深度学习模型，可实现对医疗废弃物的实时检测与分类。YOLO（You Only Look Once）系列模型因其在速度与精度之间的良好平衡，成为该类任务的理想选择。
1.2 技术选型背景
YOLO 是由华盛顿大学的 Joseph Redmon 和 Ali Farhadi 提出的一种端到端的目标检测框架，自 2015 年首次发布以来，经历了多个版本迭代。其核心思想是将目标检测视为一个回归问题，直接在图像上进行边界框和类别预测，从而实现极高的推理速度。
YOLOv5 虽非官方 Ultralytics 团队正式命名的“v5”版本（实际为社区广泛使用的称呼），但基于 PyTorch 实现的 yolov5 代码库已成为工业界最常用的检测框架之一。它具备以下优势：
- 轻量高效：支持从 yolov5n 到 yolov5x 多种尺寸模型，适用于不同算力设备
- 易用性强：提供完整的训练、验证、推理接口，支持一键式部署
- 生态完善：集成数据增强、自动锚框计算、TensorBoard 可视化等功能
本文将围绕基于 YOLOv5 的医疗废弃物识别分类系统展开，介绍如何利用预置镜像快速搭建开发环境，并完成从数据准备到模型部署的全流程实践。
2. 环境准备与镜像使用
2.1 YOLOv5 镜像简介
本项目采用 CSDN 星图平台提供的YOLO-V5 深度学习镜像，该镜像已预装以下核心组件：
- Ubuntu 20.04 LTS 操作系统
- Python 3.8 + Conda 环境管理
- PyTorch 1.13 + torchvision + CUDA 11.7 支持
- Ultralytics/yolov5 官方仓库（克隆至
/root/yolov5/） - JupyterLab、OpenCV、Pillow、matplotlib 等常用工具库
此镜像极大简化了环境配置过程，开发者可直接进入模型开发阶段，无需花费大量时间解决依赖冲突或驱动兼容性问题。
2.2 开发环境接入方式
方式一：Jupyter Notebook 使用
通过浏览器访问 JupyterLab 界面，可在交互式环境中编写和调试代码。登录后界面如下：
用户可在/work目录下创建自己的项目文件夹，或将yolov5目录复制出来进行修改：
cp -r /root/yolov5 /work/my_yolov5_project随后在 Notebook 中导入模型并测试推理功能：
方式二：SSH 远程连接
对于习惯命令行操作的开发者，可通过 SSH 工具（如 Xshell、MobaXterm）远程连接实例：
ssh root@<your-instance-ip> -p 22连接成功后即可执行训练脚本、监控 GPU 使用情况或批量处理图像数据。
3. 医疗废弃物识别系统实现
3.1 数据集构建与标注
医疗废弃物种类繁多，需根据实际应用场景定义分类体系。常见类别包括：
sharps（锐器：针头、刀片）gauze（纱布、棉球）plastic_waste（塑料制品：输液瓶、导管）medication_container（药品容器）mask（口罩、防护服残片）
我们采集了约 2000 张真实场景下的废弃物图像，涵盖不同光照、角度和遮挡条件。使用 LabelImg 工具进行矩形框标注，生成对应的.txt标签文件（每行格式：class_id center_x center_y width height，归一化坐标）。
目录结构组织如下：
dataset/ ├── images/ │ ├── train/ │ └── val/ ├── labels/ │ ├── train/ │ └── val/ └── data.yaml其中data.yaml内容示例：
train: /work/dataset/images/train val: /work/dataset/images/val nc: 5 names: ['sharps', 'gauze', 'plastic_waste', 'medication_container', 'mask']3.2 模型训练流程
进入 yolov5 项目目录并启动训练：
cd /root/yolov5/ python train.py \ --img 640 \ --batch 16 \ --epochs 100 \ --data /work/dataset/data.yaml \ --weights yolov5s.pt \ --cfg models/yolov5s.yaml \ --name medical_waste_detection关键参数说明：
--img: 输入图像分辨率--batch: 批次大小，根据显存调整--weights: 初始化权重，选用预训练模型提升收敛速度--data: 数据配置文件路径--name: 实验名称，日志保存于runs/train/medical_waste_detection
训练过程中可通过 TensorBoard 查看损失曲线、mAP 指标变化：
tensorboard --logdir runs/train3.3 推理与结果可视化
训练完成后，使用最佳模型进行推理测试：
import torch from PIL import Image import cv2 # 加载自定义训练模型 model = torch.hub.load('ultralytics/yolov5', 'custom', path='runs/train/medical_waste_detection/weights/best.pt') # 图像源可以是 URL、本地路径或 OpenCV 帧 img_path = '/work/test_samples/sharp_waste_01.jpg' results = model(img_path) # 输出检测信息 results.print() # 保存带标注的结果图 results.save(save_dir='/work/results')输出结果包含每个检测框的类别、置信度及坐标信息，例如：
medical_waste_01.jpg: 640x640 2 sharps, 1 gauze, 1 plastic_waste at 6.3ms同时生成带有彩色边框和标签的图像，便于人工审核。
4. 实践难点与优化策略
4.1 类别不平衡问题
部分类别（如“锐器”）样本较少，导致模型偏向多数类。解决方案包括：
- 过采样少数类图像：复制并轻微变换（旋转、翻转）增加样本多样性
- 引入类别权重：在损失函数中为稀有类别分配更高权重
- 使用 mAP@0.5:0.95 作为评估指标：更全面衡量模型性能
4.2 小目标检测挑战
医疗废弃物常以小尺寸出现在画面中（如远处散落的针头）。为此采取以下措施：
- 提高输入分辨率：将
--img从 640 提升至 1280（需更大显存） - 启用 Mosaic 数据增强：默认开启，有助于提升小目标泛化能力
- 调整 Anchor 尺寸：根据标注统计重新聚类生成更适合小目标的先验框
4.3 部署优化建议
为满足边缘设备部署需求，建议进行以下优化：
| 优化方向 | 具体做法 |
|---|---|
| 模型轻量化 | 使用yolov5n或yolov5s替代大模型 |
| 模型剪枝 | 移除冗余卷积层，减少参数量 |
| 量化压缩 | 将 FP32 模型转为 INT8，降低内存占用 |
| ONNX 导出 | 转换为通用格式，便于跨平台部署 |
导出 ONNX 模型命令：
python export.py --weights runs/train/medical_waste_detection/weights/best.pt --include onnx5. 总结
5.1 实践经验总结
本文基于 YOLOv5 深度学习镜像，完成了医疗废弃物识别分类系统的完整开发流程。通过合理构建数据集、调优训练参数、针对性解决小目标与类别不平衡问题，最终实现了较高准确率的自动识别能力。
核心收获如下：
- 预置镜像显著提升开发效率：省去繁琐的环境配置环节，尤其适合初学者和快速原型开发。
- 数据质量决定上限：高质量标注和多样化的图像分布是模型成功的前提。
- 轻量模型更适合落地：在保证精度的前提下，优先选择小型化模型以适应嵌入式设备。
5.2 最佳实践建议
- 建立持续更新机制：定期收集新样本并重新训练模型，应对新型废弃物出现。
- 结合规则引擎过滤误检：例如设置最小置信度阈值或空间逻辑判断（如“口罩不应出现在垃圾桶上方”）。
- 考虑隐私保护设计：若用于医院内部监控，应避免存储人脸等敏感信息，必要时添加模糊处理模块。
获取更多AI镜像
想探索更多AI镜像和应用场景？访问 CSDN星图镜像广场，提供丰富的预置镜像，覆盖大模型推理、图像生成、视频生成、模型微调等多个领域，支持一键部署。
