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