溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

Pytorch 保存模型生成圖片方式

發(fā)布時(shí)間：2020-09-21 11:58:07 來源：腳本之家 閱讀：146 作者：wzy_zju 欄目：開發(fā)技術(shù)

三通道數(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 保存模型生成圖片方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手