Pytorch 小知识点汇总一 -- save_image, Image对象size, dataloader之getitem返回值, VScode 报错’already started‘

一、torchvision.utils.save_image 报错TypeError: Cannot handle this data type

1. img如果是uint16的矩阵而不转为uint8，Image.fromarray这句会报错。所以加上np.uint8(img)很有必要

rgb_image = rgb_image.astype(np.uint8)

2. 在pytorch中tensor默认是CHW，而PIL中是HWC. 顾img需是CHW形式。

 rgb_image = rgb_image.transpose((2,0,1))

二、保存生成图片的两个方式：

1. opencv方法：

cv2.imwrite('./temp/imgIn1.png', image_temp0)

2. torchvision.utils.save_image 方法：要特别注意 除255.0  即/255.0

torchvision.utils.save_image(torch.from_numpy(image_temp0).float()/255.0, './temp/0image_'+str(index_temp)+'.jpg', nrow= 1)

三、 图片尺寸

im.size

通过Image对象的size属性可以获取图片的尺寸。图像大小，以像素为单位。大小是一个2元组(宽度、高度)(width, height)

 

<