Pytorch 保存模型生成圖片方式
更新時間：2020年01月10日 15:50:26 作者：wzy_zju
今天小編就為大家分享一篇Pytorch 保存模型生成圖片方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
三通道數(shù)組轉(zhuǎn)成彩色圖片
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ù)組轉(zhuǎn)化成灰度圖
Img_mask=np.array(img_mask)*255 ##img_mask為二值圖
Img_mask=cv2.merge(img_mask)
cv2.imwrite(img_path,img_mask)
以上這篇Pytorch 保存模型生成圖片方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
ultralytics庫由Ultralytics團隊開發(fā),旨在為YOLO系列模型提供高效、靈活且易于使用的工具,本文將詳細(xì)介紹ultralytics庫的功能、安裝方法、核心模塊以及使用示例,感興趣的朋友一起看看吧2025-03-03
這篇文章主要介紹了python自動化測試無法啟動谷歌瀏覽器問題，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下2019-10-10
今天小編就為大家分享一篇pyqt5 刪除layout中的所有widget方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
Django和websocket實現(xiàn)簡單的多人聊天的示例代碼
本文主要介紹了使用Django和WebSocket實現(xiàn)一個簡單的多人聊天應(yīng)用,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2025-01-01
Python操作MongoDB的教程詳解(插,查,改,排,刪)
MongoDB是一個基于分布式文件存儲的數(shù)據(jù)庫。是一個介于關(guān)系數(shù)據(jù)庫和非關(guān)系數(shù)據(jù)庫之間的產(chǎn)品，是非關(guān)系數(shù)據(jù)庫當(dāng)中功能最豐富，最像關(guān)系數(shù)據(jù)庫的。本文將詳細(xì)和大家聊聊Python操作MongoDB的方法，需要的可以參考一下2022-09-09
這篇文章主要介紹了Python中變量交換的例子,Python中的變量交換不需要臨時變量,一個等號即可實現(xiàn)變量的交換,非常方便,需要的朋友可以參考下2014-08-08
這篇文章主要介紹了python環(huán)境下安裝opencv庫的方法 ，本文通過實例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-03-03
Django rstful登陸認(rèn)證并檢查session是否過期代碼實例
這篇文章主要介紹了Django rstful登陸認(rèn)證并檢查session是否過期代碼實例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了Python讀取Word文檔操作，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-08-08
在本篇文章里小編給大家整理的是一篇關(guān)于python創(chuàng)建堆的方法實例講解內(nèi)容，有興趣的朋友們可以學(xué)習(xí)下。2021-03-03