Должен ли я перенести Тензор при подаче его в CNN

Question

Я использую пользовательский набор данных с изображениями разных размеров в формате Lab (Lightness, a, b), которые подаются в CNN.Входной слой имеет 3 входных канала, и поэтому я хотел разделить все 3 канала (L, a, b) и передать их в сеть.Далее мне было интересно, нужно ли транспонировать каждый тензор?Я сомневаюсь, что он потерял бы свои размеры, которые изменяются от изображения к изображению, и я не смог бы восстановить изображение в конце.Есть мысли или идеи, как мне следует нормализовать изображение?

Answer 1

Вы можете нормализовать без необходимости транспонирования изображения или разделения его по каналам

torchvision.transforms.Normalize(mean=[l_channel_mean, a_channel_mean , b_channel_mean], std= [l_channel_mean, a_channel_mean , b_channel_mean]) Единственное необходимое преобразование - это преобразование изображений в тензоры: torchvision.transforms.ToTensor()