Pytorch 如何更改调幅图像的通道尺寸？

假设我有一个火炬张量，形状为[x，y，z，21]，其中x=批量大小，y=图像宽度，z=图像高度。上面的张量代表了一批具有21个通道的图像。如何将其转换为size=[x，y，z，3]？

[x，y，z，21]->[x，y，z，1]->[x，y，z，3]

对于大小为[x，y，z，21]的分割结果

大小为[x，y，z，1]的分段类索引结果

# for pytorch, the right format for image is [batch, channels, height, width]
# however your image format [batch, height, width, channels]
result=predicts.argmax(-1)

---

索引组合颜色图将帮助您！查看详细信息

为什么有21频道图像？频道做什么？没有唯一的方法可以做到这一点。我正在使用Pascal voc 2012数据集进行语义分割，因为有21个类，每个通道代表该特定类的分数。