三通道数组转成彩色图片
1
2
3
4
5
6
7
8
9
10
11
|
img = np.array(img1) img = img.reshape( 3 ,img1.shape[ 2 ],img1.shape[ 3 ]) img = (img + 0.5 ) * 255 ##img做过归一化处理,【-0.5,0.5】 img_path = '/home/isee/wei/image/imageset/result.jpg' img = cv2.merge(img) cv2.imwrite(img_path,img) |
单通道数组转化成灰度图
1
2
3
4
5
|
Img_mask = np.array(img_mask) * 255 ##img_mask为二值图 Img_mask = cv2.merge(img_mask) cv2.imwrite(img_path,img_mask) |
以上这篇Pytorch 保存模型生成图片方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/wzy_zju/article/details/80778551