U Pha!@svddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZGdd d ejZdS) ) annotations)SequenceN) UnetOutBlock)UnetrBasicBlockUnetrPrUpBlock UnetrUpBlock)ViT)deprecated_argensure_tuple_repcsjeZdZdZeddddddd ddddddddddddddddddfdd ZddZddZZS)!UNETRz UNETR based on: "Hatamizadeh et al., UNETR: Transformers for 3D Medical Image Segmentation " pos_embedz1.2z1.4 proj_typezplease use `proj_type` instead.)namesinceremovednew_name msg_suffix convinstanceTFintzSequence[int] | intstrz tuple | strboolfloatNone) in_channels out_channelsimg_size feature_size hidden_sizemlp_dim num_headsr r norm_name conv_block res_block dropout_rate spatial_dimsqkv_bias save_attnreturncstd| krdks(ntd||dkr