文章目錄

• 一，Pytorch
• • 1. 直接保存Tensor
  • 2.Tensor 轉CV2 保存
• 二、python
• • 1. opencv
  • 2.matplotlib:
  • 3. PIL

一，Pytorch

1. 直接保存Tensor

#!/usr/bin/env python # _*_ coding:utf-8 _*_ import torch from torchvision import utils as vutilsdef save_image_tensor(input_tensor: torch.Tensor, filename):"""將tensor保存為圖片:param input_tensor: 要保存的tensor:param filename: 保存的文件名"""assert (len(input_tensor.shape) == 4 and input_tensor.shape[0] == 1)# 復制一份input_tensor = input_tensor.clone().detach()# 到cpuinput_tensor = input_tensor.to(torch.device('cpu'))# 反歸一化# input_tensor = unnormalize(input_tensor)vutils.save_image(input_tensor, filename)

2.Tensor 轉CV2 保存

二、python

1. opencv

import cv2 as cv # load `在這里插入代碼片` img = cv.imread(imagepath) # shape=(height, width, channel) h,w,c = img.shape # show cv.imshow('window_title', img) # save cv.imwrite(savepath, img)

2.matplotlib:

import matplotlib.pyplot as plt # plt 用于顯示圖片 import matplotlib.image as mpimg # mpimg 用于讀取圖片 import numpy as np# load img = mpimg.imread('cat.jpg') # 此時 img 就已經是一個 np.array 了，可以對它進行任意處理 # height, width, channel=(360, 480, 3) h,w,c = img.shape # show plt.imshow(img) # 顯示圖片 plt.axis('off') # 不顯示坐標軸 plt.show()# save # 適用于保存任何 matplotlib 畫出的圖像，相當于一個 screencapture plt.savefig('fig_cat.png')

3. PIL

from PIL import Image import matplotlib.pyplot as plt img=Image.open('/home/wanghao/Pictures/001.jpg') plt.figure("head") plt.imshow(img) plt.show()

總結

以上是生活随笔為你收集整理的python，pytorch：读取，保存，显示图片的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。