This repo provides the official code for : 1) TransBTS: Multimodal Brain Tumor Segmentation Using Transformer (https://arxiv.org/abs/2103.04430) , accepted by MICCAI2021. 2) TransBTSV2: Towards Better and More Efficient Volumetric Segmentation of Medical Images(https://arxiv.org/abs/2201.12785).
I just want to use your 3D TransUNet in your code.
but I don't know the input shape of the position parameter in forward, I use
position=torch.Tensor([1,512,2,2,2, 64])
to match the
self.pos_emb_layer = nn.Linear(6, emb_channels)
I feel I was wrong.
I just want to use your 3D TransUNet in your code. but I don't know the input shape of the position parameter in forward, I use position=torch.Tensor([1,512,2,2,2, 64]) to match the self.pos_emb_layer = nn.Linear(6, emb_channels) I feel I was wrong.