1.关于读图片
1.
img = Image.open('root')
img.save('../test.jpg')
img.show() # 显示图片
2.
img=cv2.imread('root')
plt.imshow(img,‘brg’)
pylab.show()
cv2.imwrite('root', img)
2.字符分割
string = '1_A.jpg'
a = string.split('_')[0]
b = string.split('_')[1]
3.numpy <--> tensor
tensor转换成numpy
a = torch.FloatTensor(2,9)
print a.numpy();
numpy转换成tensor
a = np.ones(10)
torch.from_numpy(a)
4.Variable <--> numpy
将Variable转换成numpy
import torch
from torch.autograd import Variable
a = Variable(torch.FloatTensor(2,5))
print a.data.numpy()
将numpy转换为Variable
import torch
import numpy as np
from torch.autograd import Variable
a = np.ones(10)
print Variable(torch.from_numpy(a))