一种改进基于YOLOv8的垃圾分类方法

本发明提供了一种改进基于YOLOv8的垃圾分类方法，包括以下步骤：S1，获取垃圾图像，并构建数据集；S2，在YOLOv8模型主干网络的下采样操作中引入SPD‑Conv模块，得到改进后的S‑YOLOv8模型；S3，将数据集输入到S‑YOLOv8模型中进行训练并评估，得到改进后的垃圾分类检测模型；S4，在使用训练好后的S‑YOLOv8模型进行推理时，引入SAHI进行切片推理。本发明通过在使用训练好后的S‑YOLOv8模型进行推理时，引入SAHI进行切片推理，在使用改进后的垃圾分类检测模型进行垃圾分类识别工作时，结合SAHI算法进行推理，能够改善小目标的垃圾的识别准确率，同时，各切片独立处理，在切片能捕获目标物体的情况下，检测质量不受影响，保证了检测质量。