numpy完全攻略


numpyで?ってなったことをまとめる。イディオム的に。

1. image.shape==[3.64,64] を [1,3,64,64]に変更する

image=np.stack([image])

2. numpyをTensor型にしてcudaに乗せる

image=torch.from_numpy(image).cuda()

3. csvから数字データを取ってきてnumpy配列に格納する

with open('name.csv','r') as f:
    reader=csv.reader(f)
    l=[row for row in reader]
    l=np.array(l).astype('int64')

csvから取ってきた後にastypeでキャストするのを忘れないようにする