python, tensorflow读取图片，保存图片

一，读取图片

img_path = 'cat_0002.jpg'
img = cv2.imread(img_path) # ndarray, 读取后图片大小为:(行数，列数，通道数）（768,1024,3）

二，保存图片

1）img图片的大小为（768,1024,3），dtype = uint8，保存img:

cv2.imwrite("img.png", img) #

2）保存中间结果，比如smap大小为（768,1024）, dtype = float32, 保存smap:

# 先归一化到0-1之间
smap_norm = (smap - smap.min()) / (smap.max() - smap.min())

# 再全部乘以255，保存
cv2.imwrite("output.png", smap_norm * 255.0)

 

小知识：

smap_norm = cv2.transpose(smap_norm, (1, 0)) # （768,1024）->（1024,768）

smap_norm = cv2.transpose(smap_norm * 255.0)) # 默认一样（768,1024）->（1024,768）