我有8张量在4D张量形状(B,C,H,W)的集合。更具体地说,我的张量的形状是(8,3,64,64),也就是说,8张图像的格式是[3x64x64]。在
我想把它们保存在1.png,2.png…,8.png等中。当我尝试torchvision.utils.save_images(my_tensor)
时,这是很好的工作。但是,这是将图像保存在单个网格中,而不是单个图像中。在
我如何解决这个问题?在
我试过下面的代码,但没用。在
for i in range(tensor.size(0)):
np_data = tensor.cpu().numpy()
imgplot = plt.imshow(np_data)
如果要保存单个图像,可以使用:
将另存为:1.png,2.png…,8.png
相关问题 更多 >
编程相关推荐