pytochはテンソル、画像、CPU、GPU、配列などの変換を実現します。

1283 ワード

pytoch テナント画像 CPU GPU 行列

1、pytouchのTensorテンソルを作成します。


torch.rand((3,224,224)) #          ，   （3,224,224）
 
torch.Tensor([3,2]) #    ，[3,2]

2，cpu上のtenssorとGPUすなわちpytouchで作成されたtensorの相互変換


b = a.cpu() # GPU → CPU
 
a = b.cuda() #CPU → GPU

3，tensorとnumpyの転化


b = a.numpy() # tensor    numpy  
 
a = b.from_numpy() # numpy     tensor

4,tochのGPU tensorは写真として保存します。


import scipy.misc
 
scipy.misc.imsave(‘pic_name',img) #img     ，  （224,224），

5,マトリクスを積み重ねてカラー画像を形成する


img = np.stack((ia,b,c),dim) #    a,b,c             dim        ，
#  a,b,c  （224,224）     ，   dim=-1，  img    （224,224,3)

6,numpy配列から画像を保存します。


from PIL import Image
 
im = Image.fromarray(A)
 
im.save("your_file.jpeg")

7，読取画像は行列です。


import matplotlib.image
im = matplotlib.image.imread('0_0.jpg')

8,保存行列は画像です。


import numpy as np
import scipy.misc
 
x = np.random.random((600,800,3))
scipy.misc.imsave('meelo.jpg', x)

以上のpytouchはテンソルtensor、写真、CPU、GPU、配列などの転換を実現しました。つまり、小編集は皆さんに共有しています。参考にしてもらいたいです。