Как добавить изображения в регистратор тензорной доски, используя write.add_image? ⇐ Python

[Anonymous]

У меня есть одноканальные изображения, и я хочу добавить их в регистратор тензорной доски с помощью Writer.add_image, как показано ниже:

Код: Выделить всё

 generated_images_to_show = torch.cat(gtimage_cpu, predimage_cpu), dim=0)
gen_image_grid = make_grid(generated_images_to_show .unsqueeze(1), nrow=inputs.size(0), padding=6, normalize=False)
writer.add_image('Validation Gen Comparison', gen_image_grid, epoch)

Мои сгенерированные изображения имеют один канал, и мне нужно построить их с помощью cmap= 'hsv', но в add_image функция cmap недоступна. Что мне делать в этом случае? Меня интересует просмотр результатов только в тензорной доске.
Одним из решений было бы преобразовать изображения в RGB-каналы, а затем использовать cmap=hsv и записать преобразованные изображения в тензорную доску, есть ли другое (возможно, более эффективное) решение?

Подробнее здесь: https://stackoverflow.com/questions/789 ... -add-image