LoginSignup
5
5

More than 3 years have passed since last update.

torchvision.transformasで画像をクロップせずにリサイズする

Last updated at Posted at 2020-02-03

関数名から、transforms.Resize()を素朴に使った方が良いのに、なぜかtransforms.RandomResizedCrop()で、強引にリサイズしていた。

オリジナル

dog.png

よく使われているやり方?

scaleとratioに(1.0, 1.0)を入力する。

transforms.Compose([
    transforms.RandomResizedCrop(size, scale=(1.0, 1.0), ratio=(1.0, 1.0)),
    transforms.ToTensor(),
])

dog_rc.png
クロップされている。

クロップされないやり方

size = (244, 244) # 縦横を揃えたい場合はtupleで入力
transforms.Compose([
    transforms.Resize(size),
    transforms.ToTensor(),
])

dog_r.png
クロップされていない!

データセットの性質によってはクロップした方が良い場合もあると思うので、ケースバイケースで。

5
5
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
5
5