由于img需要放入GPU,得到perdition,所以第一步首先把GPU中的tensor取出来放到CPU中:
img = img.cpu()#取出到cpu中
其次就是要将其转为一个“图片”,多次采用torch自带的transforms库更加好用
conver_img = transforms.Compose([transforms.ToPILImage()])
img = conver(img)
此时这个图像就可以由PIL/cv打开,但是cv打开会变成蓝色,因为cv的RGB排布方式的原因
但是这里不可以直接使用如下代码:
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
进行转换
需要将img转换成numpy才可以
img = np.array(img)
由此就可以使用cv打开原图
#禁止转载