torchvision.utils.make_grid 将一个batch的图片在一张图中显示 (torchvision.utils.save_image)

最新推荐文章于 2024-08-29 17:44:05 发布

hxxjxw

最新推荐文章于 2024-08-29 17:44:05 发布

阅读量4.3k

点赞数 1

文章标签： batch 深度学习计算机视觉

本文链接：https://blog.csdn.net/hxxjxw/article/details/122371031

版权

import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import torchvision

img = plt.imread('wave.jpg')
img_tensor = transforms.ToTensor()(img)
img_tensor = img_tensor.repeat(10,1,1,1)
img_tensor = torchvision.utils.make_grid(img_tensor)
torchvision.utils.save_image(img_tensor,'out.jpg')

可以用nrow参数，设置每行几张图片

import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import torchvision

img = plt.imread('wave.jpg')
img_tensor = transforms.ToTensor()(img)
img_tensor = img_tensor.repeat(10,1,1,1)
img_tensor = torchvision.utils.make_grid(img_tensor, nrow=4)
torchvision.utils.save_image(img_tensor,'out.jpg')

padding设置图片边距

import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import torchvision

img = plt.imread('wave.jpg')
img_tensor = transforms.ToTensor()(img)
img_tensor = img_tensor.repeat(10,1,1,1)
img_tensor = torchvision.utils.make_grid(img_tensor, nrow=4, padding=100)
torchvision.utils.save_image(img_tensor,'out.jpg')

但是这里其实有点多余，即如果使用torchvision.utils.save_image是没有必要写torch.utils.make_grid的，torchvision.utils.save_image内部会进行make_grid操作

import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import torchvision

img = plt.imread('wave.jpg')
img_tensor = transforms.ToTensor()(img)
img_tensor = img_tensor.repeat(10,1,1,1)
torchvision.utils.save_image(img_tensor,'out.jpg')