首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于CBAM YOLOv4-Mish的乳腺X线摄片肿块检测方法
引用本文:吴福彬,卢浩然,王统,徐胜舟.基于CBAM YOLOv4-Mish的乳腺X线摄片肿块检测方法[J].中南民族大学学报(自然科学版),2023(2):245-252.
作者姓名:吴福彬  卢浩然  王统  徐胜舟
作者单位:中南民族大学计算机科学学院&湖北省制造企业智能管理工程技术研究中心
基金项目:中央高校基本科研业务费专项资金资助项目(CZY22015);
摘    要:为了提高乳腺肿块的检测精度,基于YOLOv4提出了一种CBAM YOLOv4-Mish模型进行乳腺X线摄片肿块检测.该模型采用平滑、连续可导的Mish激活函数替换原模型中的Leaky ReLU激活函数,并引入了卷积块注意力模块,使模型更加关注于肿块等关键信息而忽略背景等无关信息.在DDSM数据集上的实验结果表明:CBAM YOLOv4-Mish的AP0.5为81.9%,比原始YOLOv4提升了3.6%.与其他乳腺肿块检测方法相比,该方法具有更好的检测能力.

关 键 词:乳腺X线摄片  卷积块注意力模块  乳腺肿块检测
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号