公开(公告)号：CN112270347A

公开(公告)日：2021-01-26

申请号：CN202011127578.5

申请日：2020-10-20

Applicant: 西安工程大学

Inventor： 管声启 ,  倪奕棋 ,  卫艳芳 ,  王俊强 ,  常江 ,  胡璐萍 ,  王琪璇 ,  张潇

IPC: G06K9/62 ,  G06N3/04

Abstract: 本发明公开了一种基于改进SSD的医疗废弃物分类检测方法，包括：采集图像样本，对图像样本进行预处理；对预处理后的每张图像样本进行标注，建立图像数据集，并将图像数据集分为训练集、验证集、测试集；将SSD模型中的VGG‑16网络替换为MobileNetV3‑Large网络，去除平均池化层，并使用多尺度特征提取网络代替末端的1×1卷积层，将每个多特征倒残差层与其前、后一层或者多层倒残差层输出特征输入特征融合模块；将四个3×3卷积层的输出特征分别输入卷积注意力模块；建立目标损失函数，设置训练参数，将训练集输入改进SSD模型进行训练，得到改进SSD模型；利用验证集、测试集对改进SSD模型进行验证、测试，得到图像样本的分类检测结果。