PyTorchで、画像の回転を、学習中の任意のタイミングで行う方法を教えてください。
backpropagationを行えるようにしたいので、numpyを通さずに、Pytorchの関数のみで行いたいです。
torchvision.transforms.functional.rotateを使えば良いでしょうか?
その場合、画像をtorch.Tensor型をPIL Image型に変換しなければならないと思いますが、
その方法も教えてください。
torchvision.transforms.functional.to_pil_image(image)とすると
PIL Imageに変換できるようですが、
Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.
というエラーが出て、結局numpy型が引数として要求されるようです。
PIL Image型に変換しないで画像の回転を行う方法はあるでしょうか?
あなたの回答
tips
プレビュー