チラ裏備忘録

情報整理

2021-01-01から1年間の記事一覧

int64型のnumpy配列では,ToTensor()による正規化が行われない?

型をuint8にすると正しく[0, 1]になりました.以後,気をつけます. import numpy as np import torchvision.transforms as transforms t = transforms.ToTensor() a = np.random.randint(0, 255, size=(4, 3)) print(a.dtype) print(t(a)) # なぜか[0, 1]に…

nn.○○とnn.functional.○○の違い

qiita.com nn.ReLU()は、nn.Moduleを作ります。つまり、nn.Sequential()に追加できます。反対にnn.functional.reluは、relu関数を呼ぶだけで、もし記述するならforward methodの方に書きます。好きな方を使えば良いみたいですが、個人的には、print(model)と…

Adaptive_avg_pool2dとavg_pool2dの違い

discuss.pytorch.org test_input = Image.open('test_image.jpeg') # 1024x768 plt.figure(figsize=(12,4)) # avg_pool2d outputs1 = F.avg_pool2d(torchvision.transforms.ToTensor()(test_input), kernel_size=(4, 4), stride=2, padding=1) print(outputs…