今天小编就为大家分享一篇Pytorch 保存模型生成图片方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

三通道数组转成彩色图片 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) 单通道数组转化成