问题描述
我正在做语义分割。我的模型输出形状为(N,C,H,W)的图像。对于VOC2012数据集，我有21个类，所以C = 21。
我想将输出绘制为单个图像，每个通道的颜色不同。有人可以告诉我可视化此图像的最佳方法是什么？我觉得我将不得不在各个通道上做一个argmax来确定哪个通道需要显示在一个像素上。但是我不确定之后该怎么办。
我的回购：VOC-Semantic-Segmentation
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）
