TransUnet如何进行多分类图片输出

Question

问题遇到的现象和发生背景

     有偿咨询，我不是计算机专业的，最近在修改transunet的代码处理我们专业的数据。训练集的数据不是图片，是csv文件，data和label都是单通道的。data是波形数据，范围是0-1；label是地质模型，数据是整数，但是有多个不同的值。
    但是我发现用TransUnet每一通道的输出数据只能做二分类的切割，只能区分归一化后的0和1，如果一个输出模型有多个参数或者不是0-1就毫无效果。
    想咨询一下这个代码怎么修改可以在一张输出图片上显示多分类的结果呢？临近毕业这个问题解决不了，如果能给我修改的方向，真的真的很感谢博主！！！！！！

用代码块功能插入代码，请勿粘贴截图

我想要达到的结果

可以在一个输出通道上同时显示多分类的训练结果。

Answer 1

TransUnet是实例分割模型，本身就支持多个类别的结果分割，你这个应该是作者改了输出或者类别就两类才会这样，你可以看下原来的网络，支持多类别分割的。

GitHub - Beckschen/TransUNet: This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation. This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation. - GitHub - Beckschen/TransUNet: This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation. https://github.com/Beckschen/TransUNet