Pytorch保存模型⽣成图⽚⽅式
三通道数组转成彩⾊图⽚
img=np.array(img1)
shape(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.imwrite(img_path,img)
单通道数组转化成灰度图
Img_mask=np.array(img_mask)*255 ##img_mask为⼆值图
Img_(img_mask)
cv2.imwrite(img_path,img_mask)
以上这篇Pytorch 保存模型⽣成图⽚⽅式就是⼩编分享给⼤家的全部内容了,希望能给⼤家⼀个参考,也希望⼤家多多⽀持。python新手代码图案如何保存

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。