一种基于改进YOLOv7模型的结直肠息肉检测与分类方法

    公开(公告)号:CN118038161A

    公开(公告)日:2024-05-14

    申请号:CN202410208741.2

    申请日:2024-02-26

    Abstract: 本发明提出一种基于改进YOLOv7模型的结直肠息肉检测与分类方法,包括:S1.获取结直肠息肉数据集;S2.对结直肠息肉数据集进行数据预处理;S3.改进YOLOv7模型,在YOLOv7的backbone网络中添加CBAM注意力机制模块,提高网络对结直肠息肉重要特征的表示能力;S4.将backbone网络中的最后一个ELAN模块中的部分普通卷积替换为可变核卷积AKConv,在提高模型性能的同时实现轻量化;S5.将YOLOv7的head网络中的RepConv替换为CoordConv,提高模型感知结直肠息肉位置信息的能力;S6.将边界框损失函数CIoU替换为Shape‑IoU损失函数,得到最终改进的YOLOv7模型;本发明设计了一种基于改进YOLOv7模型的结直肠息肉检测与分类方法,有效提高了结直肠息肉检测与分类的精确度和效率,实现了模型轻量化,对于临床诊断有重要意义。

Patent Agency Ranking