В image_tensor хранится изображение (image_tensor размера (3,256,512), в котором хранятся значения в интервале 0,255), которое я хотел бы отобразить в Tensorboard (более конкретно, TensorboardX для PyTorch) через add_image () функция для SummaryWriter. Когда я добавляю изображение в Tensorboard через writer.add_image("color_image",image_tensor,self.step), цвета инвертируются.
Когда я записываю изображение в файл с помощью scipy.misc.imsave("/write/to/path/image.png",np.transpose(image_tensor.data.cpu().numpy(),(1,2,0))), изображение прекрасно.
Единственное, что я изменяю для второй строки, это изменение CxHxW на HxWx C, но я не думайте, что это root этой проблемы с инверсией цвета. В чем может быть проблема?