import torchvision.utils as vutils
针对NCHW 的torch保存N组图像
需要在之前归一化到0-1
vutils.save_image(save_img, 'imge.png',nrow=5)
import torchvision.utils as vutils
针对NCHW 的torch保存N组图像
需要在之前归一化到0-1
vutils.save_image(save_img, 'imge.png',nrow=5)