一种基于金字塔视觉转换器的膀胱肿瘤分割方法

专利主权项内容

1.一种基于金字塔视觉转换器的膀胱肿瘤分割方法，其特征在于，包括：步骤(1)：采用PVTv2提取待分割图像在四个不同层次下的语义特征，并将底层语义特征输入LDA模块，对所述底层语义特征进行净化与校准，得到第一特征图，将三个高级语义特征分别输入MSCA模块，提取三个所述高级语义特征各自的多尺度特征信息，得到第二特征图、第三特征图、第四特征图；步骤(2)：将所述第二特征图、所述第三特征图、所述第四特征图共同输入DAM模块，进行所述高级语义特征的逐层相乘融合，得到第五特征图；步骤(3)：调节所述第一特征图的通道数和尺寸与所述第五特征图一致，并与所述第五特征图共同输入FEM模块，采用全局注意力将所述底层语义特征的语义细节信息嵌入所述高级语义特征中，得到第六特征图；步骤(4)：将所述第五特征图和所述第六特征图分别输入1×1卷积层，并插值到与所述待分割图像相同大小，得到初步分割预测结果和最终分割预测结果，并采用加权二元交叉熵损失与Dice损失结合的形式分别计算所述初步分割预测结果以及所述最终分割预测结果的损失，作为总损失。