search

Rubics-Xuan / TransBTS

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).
Apache License 2.0
388 stars 81 forks source link

A bug for the input #25

Open zhang-qiang-github opened 3 years ago

zhang-qiang-github commented 3 years ago

In the TransBTS_downsample8x_skipconnection.TransformerBTS:

self.Unet = Unet(in_channels=4, base_channels=16, num_classes=4)

This code would force the input has 4 channel. I think it should be modified as:

self.Unet = Unet(in_channels=num_channels, base_channels=16, num_classes=4)

Am I right?

cqlouis commented 3 years ago

My point :
for the Brats Dataset, the input does have 4 channel, ie. t1, t1ce, t2, flair,
so I think it's correct, no bug.

zhang-qiang-github commented 3 years ago

For Brats Dataset, it is OK. But, for another dataset, the channel may be not 4. So, self.Unet = Unet(in_channels=num_channels, base_channels=16, num_classes=4) may be more robust.

cqlouis commented 3 years ago

For Brats Dataset, it is OK. But, for another dataset, the channel may be not 4. So, self.Unet = Unet(in_channels=num_channels, base_channels=16, num_classes=4) may be more robust.

yes， agree with you