中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

Pytorch中Tensor如何與各種圖像格式相互轉化

發布時間:2021-08-12 10:28:17 來源:億速云 閱讀:376 作者:小新 欄目:開發技術

這篇文章給大家分享的是有關Pytorch中Tensor如何與各種圖像格式相互轉化的內容。小編覺得挺實用的,因此分享給大家做個參考,一起跟隨小編過來看看吧。

格式轉換

我們一般在pytorch或者python中處理的圖像無非這幾種格式:

  • PIL:使用python自帶圖像處理庫讀取出來的圖片格式

  • numpy:使用python-opencv庫讀取出來的圖片格式

  • tensor:pytorch中訓練時所采取的向量格式(當然也可以說圖片)

注意,之后的講解圖片格式皆為RGB三通道,24-bit真彩色,也就是我們平常使用的圖片形式。

PIL與Tensor

PIL與Tensor的轉換相對容易些,因為pytorch已經提供了相關的代碼,我們只需要搭配使用即可:

所有代碼都已經引用了(之后的代碼省略引用部分):

import torch
from PIL import Image
import matplotlib.pyplot as plt

# loader使用torchvision中自帶的transforms函數
loader = transforms.Compose([
  transforms.ToTensor()]) 

unloader = transforms.ToPILImage()

1 PIL讀取圖片轉化為Tensor

# 輸入圖片地址
# 返回tensor變量
def image_loader(image_name):
  image = Image.open(image_name).convert('RGB')
  image = loader(image).unsqueeze(0)
  return image.to(device, torch.float)

2 將PIL圖片轉化為Tensor

# 輸入PIL格式圖片
# 返回tensor變量
def PIL_to_tensor(image):
  image = loader(image).unsqueeze(0)
  return image.to(device, torch.float)

3 Tensor轉化為PIL圖片

# 輸入tensor變量
# 輸出PIL格式圖片
def tensor_to_PIL(tensor):
  image = tensor.cpu().clone()
  image = image.squeeze(0)
  image = unloader(image)
  return image

4 直接展示tensor格式圖片

def imshow(tensor, title=None):
  image = tensor.cpu().clone() # we clone the tensor to not do changes on it
  image = image.squeeze(0) # remove the fake batch dimension
  image = unloader(image)
  plt.imshow(image)
  if title is not None:
    plt.title(title)
  plt.pause(0.001) # pause a bit so that plots are updated

5 直接保存tensor格式圖片

def save_image(tensor, **para):
  dir = 'results'
  image = tensor.cpu().clone() # we clone the tensor to not do changes on it
  image = image.squeeze(0) # remove the fake batch dimension
  image = unloader(image)
  if not osp.exists(dir):
    os.makedirs(dir)
  image.save('results_{}/s{}-c{}-l{}-e{}-sl{:4f}-cl{:4f}.jpg'
        .format(num, para['style_weight'], para['content_weight'], para['lr'], para['epoch'],
            para['style_loss'], para['content_loss']))

numpy與Tensor

numpy格式是使用cv2,也就是python-opencv庫讀取出來的圖片格式,需要注意的是用python-opencv讀取出來的圖片和使用PIL讀取出來的圖片數據略微不同,經測試用python-opencv讀取出來的圖片在訓練時的效果比使用PIL讀取出來的略差一些(詳細過程之后發布)。

之后所有代碼引用:

import cv2
import torch
import matplotlib.pyplot as plt

numpy轉化為tensor

def toTensor(img):
  assert type(img) == np.ndarray,'the img type is {}, but ndarry expected'.format(type(img))
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
  img = torch.from_numpy(img.transpose((2, 0, 1)))
  return img.float().div(255).unsqueeze(0) # 255也可以改為256

tensor轉化為numpy

def tensor_to_np(tensor):
  img = tensor.mul(255).byte()
  img = img.cpu().numpy().squeeze(0).transpose((1, 2, 0))
  return img

展示numpy格式圖片

def show_from_cv(img, title=None):
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
  plt.figure()
  plt.imshow(img)
  if title is not None:
    plt.title(title)
  plt.pause(0.001)

展示tensor格式圖片

def show_from_tensor(tensor, title=None):
  img = tensor.clone()
  img = tensor_to_np(img)
  plt.figure()
  plt.imshow(img)
  if title is not None:
    plt.title(title)
  plt.pause(0.001)

注意

上面介紹的都是一張圖片的轉化,如果是n張圖片一起的話,只需要修改一下相應代碼即可。

舉個例子,將之前說過的修改略微修改一下即可:

# 將 N x H x W X C 的numpy格式圖片轉化為相應的tensor格式
def toTensor(img):
  img = torch.from_numpy(img.transpose((0, 3, 1, 2)))
  return img.float().div(255).unsqueeze(0)

感謝各位的閱讀!關于“Pytorch中Tensor如何與各種圖像格式相互轉化”這篇文章就分享到這里了,希望以上內容可以對大家有一定的幫助,讓大家可以學到更多知識,如果覺得文章不錯,可以把它分享出去讓更多的人看到吧!

向AI問一下細節

免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。

AI

灯塔市| 寿宁县| 峨边| 丹凤县| 精河县| 双桥区| 邯郸县| 运城市| 寻甸| 格尔木市| 禄丰县| 宿迁市| 云林县| 罗江县| 怀柔区| 类乌齐县| 平江县| 西峡县| 崇左市| 惠东县| 张家口市| 盐边县| 板桥市| 同心县| 龙海市| 石门县| 吴江市| 白银市| 新竹市| 堆龙德庆县| 山西省| 普定县| 晴隆县| 罗田县| 米脂县| 门头沟区| 马龙县| 太白县| 华亭县| 綦江县| 内黄县|