改进YOLOv7算法的龋齿图像检测

打开文本图片集

摘  要： 针对口腔医疗资源紧缺和龋齿治疗效率不足的问题，提出一种改进YOLOv7的龋齿图像检测算法，旨在协助医生进行更有效的医疗诊断，同时增强患者对预防龋齿的意识。首先，在YOLOv7算法的主干网络引入ECA⁃MobileOne网络模块代替原有的ELAN模块，降低模型参数量，提高对小目标龋齿特征的有效提取；其次，在特征图输出层采用自适应特征融合（ASFF），自适应地学习各尺度特征图在融合时的空间权重，充分获取口腔图像中不同尺度下的关键特征，提高检测的全局性和准确性；另外，采用soft⁃NMS算法替换原有的非极大值抑制算法（NMS），在牙齿异位或重叠等情况下能更有效地提升检测效果。（剩余15470字）