今天把tensor转化为图片来看看,结果怎么变成荧光色了,我想要彩色的,shape也是三维通道没问题啊。
一定是我太菜了,在那里搞了各种
1.是不是本身就是灰度
2.是不是通道顺序没对
3.是不是转tensor弄得PIL好像会变颜色
最后解决却是没有归一化!把那个tensor/255就好啦
for b in examples: #example是一个list
for j in b: #b也是list
img_t_i = torch.tensor(j[2]) #我只要第2个维度的tensor图片
img_t_i = img_t_i/255
# print(np.shape(b)) #torch.Size([3, 128, 128])
img_t_i = tensor_to_PIL(img_t_i)
Img_Name = "./template/" + str(count) + ".png" # 这个是文件命名规则 “.jpg“可以修改成其他格式,”./template/“文件夹路径
print(type(img_t_i))
image = np.asarray(img_t_i)
print(image.shape)
cv2.imwrite(Img_Name, image) #报错要么转成Numpy array但是就是灰度图片
count = count + 1
#cv2.waitkey(0)
#print(j[2])