関数名から、transforms.Resize()
を素朴に使った方が良いのに、なぜかtransforms.RandomResizedCrop()
で、強引にリサイズしていた。
オリジナル
よく使われているやり方?
scaleとratioに(1.0, 1.0)を入力する。
transforms.Compose([
transforms.RandomResizedCrop(size, scale=(1.0, 1.0), ratio=(1.0, 1.0)),
transforms.ToTensor(),
])
クロップされないやり方
size = (244, 244) # 縦横を揃えたい場合はtupleで入力
transforms.Compose([
transforms.Resize(size),
transforms.ToTensor(),
])
データセットの性質によってはクロップした方が良い場合もあると思うので、ケースバイケースで。