发明公开
- 专利标题: 一种基于残差CBAM注意力机制的肿瘤图像分类方法及装置
-
申请号: CN202311374374.5申请日: 2023-10-23
-
公开(公告)号: CN117611872A公开(公告)日: 2024-02-27
- 发明人: 刘强 , 刘新锐 , 滕达 , 王文通 , 卢思超
- 申请人: 北京石油化工学院
- 申请人地址: 北京市大兴区清源北路19号
- 专利权人: 北京石油化工学院
- 当前专利权人: 北京石油化工学院
- 当前专利权人地址: 北京市大兴区清源北路19号
- 代理机构: 北京细软智谷知识产权代理有限责任公司
- 代理商 刘迪
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06T7/00
摘要:
本申请提供一种基于残差CBAM注意力机制的肿瘤图像分类方法及装置,属于医学图像分类技术领域。方法先获取脑部CT图像集;然后基于残差CBAM注意力机制,采用跳跃连接方式构建残差注意力模块;将残差注意力模块嵌入ResNet18残差网络模型的残差结构,并将残差结构的平均池化层修改为全局平均池化层,获得改进的ResNet18残差网络模型;最后将脑部CT图像集输入改进的ResNet18残差网络模型进行分类训练,并将训练后的模型确定为肿瘤图像分类模型。本发明通过在肿瘤图像分类模型中添加残差注意力模块,可以使得模型提取更具有分辨性的特征,提高了对肿瘤图像分类的准确率。