1. 直接保存Tensor

#!/usr/bin/env python # _*_ coding:utf-8 _*_ import torch from torchvision import utils as vutilsdef save_image_tensor(input_tensor: torch.Tensor, filename):"""將tensor保存為圖片:param input_tensor: 要保存的tensor:param filename: 保存的文件名"""assert (len(input_tensor.shape) == 4 and input_tensor.shape[0] == 1)# 復(fù)制一份input_tensor = input_tensor.clone().detach()# 到cpuinput_tensor = input_tensor.to(torch.device('cpu'))# 反歸一化# input_tensor = unnormalize(input_tensor)vutils.save_image(input_tensor, filename)

Tensor 轉(zhuǎn)CV2 保存

總結(jié)

以上是生活随笔為你收集整理的Pytorch：保存图片的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。