更換前,要用nnUNet V2跑通所用數據集,證明nnUNet V2、數據集、運行環境等沒有問題
閱讀nnU-Net V2 的 U-Net結構,初步了解要修改的網絡,知己知彼,修改起來才能游刃有余。
MultiRes Block 是 MultiResUNet 中核心組件之一,旨在解決傳統 U-Net 在處理多尺度醫學圖像時的局限性。傳統的 U-Net 使用固定大小的卷積核(如 3x3),這在處理具有不同尺度特征的醫學圖像時可能不夠靈活。MultiRes Block 通過引入多尺度卷積操作來捕捉不同尺度的特征:它采用了類似于 Inception 模塊的思想,使用不同大小的卷積核(如 3x3、5x5 和 7x7)并行處理輸入特征圖,然后將這些特征圖拼接在一起。為了減少計算開銷,MultiRes Block 將較大的卷積核(如 5x5 和 7x7)分解為多個 3x3 卷積層的堆疊。MultiRes Block 還引入了殘差連接。
圖3圖注機翻:開發提出的MultiRes Block。我們從一個簡單的類Inception塊開始,通過并行使用3×3、5×5和7×7卷積濾波器,并將生成的特征圖進行拼接(圖3(a))。這使得我們能夠從不同上下文大小中調和空間特征。我們沒有并行使用3×3、5×5和7×7濾波器,而是將較大且計算代價較高的5×5和7×7濾波器分解為一系列3×3濾波器的連續操作(圖3(b))。圖3?展示了MultiRes Block,其中我們逐漸增加了連續三層中的濾波器數量,并添加了殘差連接(以及1×1濾波器以保持維度)。
MultiResUNet :官方代碼