Я работаю над моделью pytorch, которая выводит изображение. Проблема заключается в отображении этого изображения, поскольку на выходе получается тензор факела со значениями от 0 до 1. Если я покажу это так, он будет работать нормально
out=model.forward(x)
out=torch.squeeze(out)
out=out.permute(1,2,0)
plt.imshow(out)
EDIT: Вот как выглядел этот результат, это не фактический результат, поскольку я потерял веса сейчас, но одинаково для всех целей этого вопроса
Однако, если я конвертирую его в изображение PIL перед отображением, оно дает странные цветные пятна, как показано
out=model.forward(x)
out=torch.squeeze(out)
out=transforms.ToPILImage()(out)
plt.imshow(out)
Я не использую первый метод, так как после этого у меня есть еще несколько шагов, которые требуют, чтобы у меня было изображение с целочисленными значениями от 0 до 255.