Pytorch 保存模型生成圖片方式
更新時間：2020年01月10日 15:50:26 作者：wzy_zju
今天小編就為大家分享一篇Pytorch 保存模型生成圖片方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
三通道數(shù)組轉成彩色圖片
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)
單通道數(shù)組轉化成灰度圖
Img_mask=np.array(img_mask)*255 ##img_mask為二值圖
Img_mask=cv2.merge(img_mask)
cv2.imwrite(img_path,img_mask)
以上這篇Pytorch 保存模型生成圖片方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Python入門教程之Python的安裝下載配置,Python是一門非常強大好用的語言,也有著易上手的特性,本文為入門教程,需要的朋友可以參考下2023-04-04
這篇文章主要介紹了關于Python中的海象運算符“:=”使用方法詳解,海象運算符（walrus?operator）是?Python?3.8?中引入的一種新的語法,需要的朋友可以參考下2023-04-04
這篇文章主要介紹了關于DataFrame中某列值的替換map(dict),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
這篇文章主要介紹了Python面向對象的程序設計詳情，面向對象的程序設計在Python中具有非常重要的地位，熟練的使用面向對象編程能夠為我們的Python編程提供很多的便利之處，希望您閱讀完本文后能夠有所收獲2022-01-01
這篇文章主要為大家詳細介紹了如何利用Python實現(xiàn)24點游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-12-12
這篇文章主要介紹了pytorch配置雙顯卡方式,使用雙顯卡跑代碼,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-06-06
這篇文章主要講解了Python爬蟲的基礎知識，文中相關知識講解的非常詳細，感興趣的小伙伴可以參考參考2021-08-08
今天小編就為大家分享一篇python讀取文件名并改名字的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了Python colormap庫的安裝和使用詳情，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-10-10
這篇文章主要介紹了如何基于python實現(xiàn)畫不同品種的櫻花樹,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-01-01