Giters

Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging

Home Page:https://monai.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Support for pre-activation residual blocks in SwinUNETR

philnovv opened this issue · comments

commented

Is your feature request related to a problem? Please describe.
SwinUNETR v2 (https://rdcu.be/dnwDO) suggest the use of pre-activation residual blocks. As it stands, the MONAI implementation of SwinUNETR only accommodates post-activation residual blocks (setting res_block = True when calling UnetrBasicBlock).

Describe the solution you'd like
An option to specify whether pre- or post-activation should be used when initializing UnetrBasicBlock.
UnetrBasicBlock calls monai.networks.blocks.dynunet_block.UnetResBlock during init; the latter can be easily changed to accept an additional str input specifying the type of residual block required.