一种基于YOLOv5模型的牙齿图像检测装置和方法

本发明公开了一种基于YOLOv5模型的牙齿图像检测装置和方法，应用于医学图像领域，包括：获取预先训练好的牙齿图像检测模型；牙齿图像检测模型在YOLOv5卷积神经网络为架构的基础上引入了注意力机制，并且将HardSwish激活函数作为激活函数，将EIOU损失函数作为模型训练过程中的损失函数；将待检测的牙齿图像输入至训练好的牙齿图像检测模型，检测是否存在病灶区域；若存在，则病灶区域通过矩形框标注的方式进行显示，并显示病灶信息；病灶信息至少包括病灶种类、数量、尺寸和位置信息。本发明提高了YOLOv5卷积神经网络检测牙齿图像的速度和精度，检测效果上得到明显提升。