torchvision.transformsでランダムに画像を90°単位で回転させる - Qiita

はじめに torchvision.transformsで画像を90°単位でランダム回転させたい！ そう思ったのが事の始まりです。 RandomRotationじゃダメなの？ torchvision.transformsでランダム回転をさせるにはRandomRotationがあるのですが、90°単位で回転させるとなるとちょっと痒い所に手が届かない感じ。 主に次の二点が問題。 指定した角度の範囲内でランダムに回転させるものとなっていること ※角度の範囲指定を(90, 90)とかにすればできなくはないのですが、90°, 180°, 270°でそれぞれ用意する必要があるためあまりスマートではないですよね。 回転対象がTensor型の場合、その配列がPILで読み込める(例えば配列中身のデータ型がint8である)ものでないとエラーになってしまうこと ※例えば地理空間データとかを画像っぽくCNNで扱うと