一种外阴疾病分类的神经网络模型及其训练方法

本发明公开了一种外阴疾病分类的神经网络模型及其训练方法，该模型包括：图像获取模块，用于获取外阴图像，并将所述外阴图像输入Swin Transformer模块；Swin Transformer模块，用于将所述外阴图像分割为多个不重叠的固定尺寸图块，并将各所述图块转化为图块序列，通过注意力机制对所述图块序列进行层次化多尺度的特征提取，得到多尺度层次化的特征图，对所述特征图进行外阴疾病类别预测，输出预测的外阴疾病类别以及对应概率；本发明通过外阴疾病分类的神经网络模型，实现了从图像输入到疾病预测的全流程自动化，避免了人工诊断的主观偏差，提高了对外阴疾病诊断分类的准确性。