摘要
本发明公开了一种基于改良Swin Transformer模型的肺炎图像识别分类方法,包括以下步骤:S100.对原始数据进行内容标准化、图像色彩化和格式统一化三个方面的预处理;S200.对输入训练集图像进行数据增强;S300.选择ResNet‑34作为教师模型,并使用教师模型在处理好的训练集上进行提前训练,得到用以指导学生模型的硬标签;S400.使用改进后的Swin Transformer学生模型进行训练,利用包括但不限于MSG Token与shuffle的方法提高学生模型训练效率;并在计算损失函数时,利用教师模型输出的硬标签对学生模型进行指导;S500.导入待分类的胸片影像,利用训练好的学生模型得到分类结果。本发明可解决现有深度学习模型在处理大规模数据时存在的泛化能力不足、计算复杂度高以及对图像局部特征提取不足等问题。
技术关键词
图像识别分类方法
学生
教师
图像局部特征提取
注意力
标签
图片
全局平均池化
图像特征提取
优化器
训练集数据
深度学习模型
像素
影像
格式
参数
色彩
系统为您推荐了相关专利信息
时空分布图
积水
背景噪声干扰
上下文特征
监控视频图像
室内场景图像
补全方法
像素点
注意力机制
语义特征
情感分类模型
情感分类方法
分类网络
情感类别
训练集
早期预警方法
多任务
叶面积指数
无人机高光谱影像
通道注意力机制