CV・NLPハマりどころメモ

画像認識と自然言語処理を研究する上でうまくいかなかったことと,その対策をまとめる自分用のメモが中心.

Pytorch

TensorDatasetで生画像をTensor化して読み込む[PyTorch]

from glob import glob from PIL import Image import numpy as np import torch import torch.utils.data as data_utils from torchvision.transforms import ToTensor path = "./imgs/*" files = glob(path) labels = np.array([1,0,1,1,0]) labels = torc…

Pytorch v0.4のコードをv0.3で動かす際には.dataに注意!![Pytorch]

Pytorchのコードを見ているとミニバッチごとのlossやaccuracyを計算する際、.dataを用いて値を取り出されることが頻繁にある。 よくある例: for i in range(0, 2 * POS_NEG_SAMPLES, BATCH_SIZE): inp, target = dis_inp[i:i + BATCH_SIZE], dis_target[i:i …

PytorchでGPUが使えているかを確認するコマンド

import torch print(torch.cuda.is_available()) >> True torch.cuda.get_device_name(0) >> 'GeForce GTX 1080 Ti'