Pytorch 小知识点汇总

一、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