這篇文章主要用實例分析pytorch讀取圖像數(shù)據(jù)如何轉成opencv格式,內容清晰明了,對此有興趣的小伙伴可以學習一下,相信大家閱讀完之后會有幫助。
pytorch讀取圖像數(shù)據(jù)轉成opencv格式方法:先轉成numpy通用的格式,再將其轉換成opencv格式。
pytorch讀取的數(shù)據(jù)使用loaddata這類函數(shù)實現(xiàn)。pytorch網(wǎng)絡輸入圖像的格式為(C, H, W),就是(通道數(shù),高,寬)而numpy中圖像的格式為(H,W,C)。
那就將其通道調換一下。用到函數(shù)transpose。
轉換方法如下
例如A 的格式為(c,h,w) 那么經(jīng)過
A = A.transpose(1,2,0)
后就變成了(h,w,c)了
然后用語句
B= cv2.cvtColor(A,cv2.COLOR_RGB2BGR)
結果就變成opencv可用的圖像了。 如果不做transpose轉換,那么得到的圖像是一個1*h大小的圖.......
完整代碼:
變換部分:
一般的pytorch會進行裁剪 放縮 歸一化等操作。例如
transforms = Compose([ ToTensor(),//將數(shù)據(jù)除以255加載進來 Resize(768),//裁剪768*768大小的圖像 ConvertMaskID(Cityscapes.classes),//與這個事無關不用去管它 Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])// 歸一化 利用均值方差歸一化 ])
另外有需要云服務器可以了解下創(chuàng)新互聯(lián)scvps.cn,海內外云服務器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢,專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應用場景需求。
本文名稱:用實例分析pytorch讀取圖像數(shù)據(jù)如何轉成opencv格式-創(chuàng)新互聯(lián)
當前路徑:http://jinyejixie.com/article4/hesie.html
成都網(wǎng)站建設公司_創(chuàng)新互聯(lián),為您提供關鍵詞優(yōu)化、網(wǎng)站收錄、外貿網(wǎng)站建設、網(wǎng)站內鏈、網(wǎng)站設計公司、標簽優(yōu)化
聲明:本網(wǎng)站發(fā)布的內容(圖片、視頻和文字)以用戶投稿、用戶轉載內容為主,如果涉及侵權請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內容未經(jīng)允許不得轉載,或轉載時需注明來源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內容