新一代信息技術
基于改進的EfficientNetV2網絡的腦腫瘤分類方法
本發明提出了一種基于改進的EfficientNetV2網絡和注意力機制的腦腫瘤圖像分類方法,采用一種基于EfficientNetV2網絡模型改進的腦腫瘤圖像分類方法,在基礎的EfficientNetV2網絡結構的第一層3x3卷積層后加入殘差結構,通過在殘差結構中加入三層3x3的卷積層,一層ECA注意力機制,一層1x1卷積層,提取特征信息。同時在前三層的Fused?MBConv模塊的shortcut連接中加入CMAB注意力機制,使網絡更加關注淺層網絡中的關鍵信息,最后將殘差結構的輸出結果和主干網絡最后一層的MBConv模塊的輸出結果相加,在保證網絡深度的同時更好的提取淺層與深層的特征信息,進行特征融合,提升分類網絡性能。
長春工業大學
授權發明