Сжать размерность тензора с 3 до 1
Есть матрица картинок. [Count, 3, 493, 733] (Количество,RGB,H,W)
Картинки переведены при помощи команды Grayscale в оттенки серого. Но размерность не поменялась. Shape по прежнему выдаёт - [Count, 3, 493, 733]. Как мне получить [Count, 1, 493, 733]??? Подскажите пожалуйста.
Ответы (1 шт):
Автор решения: Andrew
→ Ссылка
Согласно документации:
Grayscale version of the input.
If num_output_channels == 1 : returned image is single channel
If num_output_channels == 3 : returned image is 3 channel with r == g == b
Предположу, что в коде необходимо переопределить количество выходных каналов:
Grayscale(num_output_channels=1)