Matplotlib 在语义分割中，通过为每个通道指定颜色来打印图像
我正在做语义分割。我的模型输出一个形状的图像Matplotlib 在语义分割中，通过为每个通道指定颜色来打印图像,matplotlib,autoencoder,cnn,semantic-segmentation,Matplotlib,Autoencoder,Cnn,Semantic Segmentation,我正在做语义分割。我的模型输出一个形状的图像（N、C、H、W）。对于VOC2012数据集，我有21个类，因此C=21 我想将输出绘制为单个图像，每个通道使用不同的颜色。有人能告诉我什么是最好的方式来可视化这个图像吗？我觉得我必须在各个通道之间做一个argmax，以确定哪个通道需要以像素显示。但我不知道那之后该怎么办 我的回购协议：
（N、C、H、W）