Как изменить размер канала изображения?

Question

Предположим, у меня есть тензор факела с shape = [x,y,z,21], где x = batch_size, y = image_width, z= image_height. Вышеуказанный тензор представляет собой серию изображений с 21 каналом. Как мне преобразовать его в size = [ x,y,z,3 ]?

Answer 1

[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)

Индексная комбинация цветов поможет вам! просмотр карта цветов для detial