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

Why setting num_classes+1 as the output classes? #39

Open johnson111788 opened 2 years ago

johnson111788 commented 2 years ago

I notice that the output of the model contains 4 classes, but there's only 3 classes of tumor. I'm wondering what is the purpose of the dimension zero?