在训练或者别的过程中储存图像的方法
此处省略数据转换和载入dataset的步骤
from torchvision.utils import save_image
img_num=0
for _ in range(10):
for img,label in dataset:
save_image(img,'img'+str(img_num)+'.png')
img_num+=1
在训练或者别的过程中储存图像的方法
此处省略数据转换和载入dataset的步骤
from torchvision.utils import save_image
img_num=0
for _ in range(10):
for img,label in dataset:
save_image(img,'img'+str(img_num)+'.png')
img_num+=1