原本想要torchvision.transforms.RandomRotation进行图片旋转
官网地址:Transforming and augmenting images — Torchvision 0.12 documentation
发现需要输入PIL图片或者tensor的格式为(...,H,W)
且是随机旋转
比如:transform = transforms.RandomRotation(90)代表在-90°到90°之间随机旋转某一个角度;
transform = transforms.RandomRotation(degrees=[90,180])代表在90°到180°之间随机旋转某一个角度。坑在于不像我们需要的旋转90\180\270
如果需要旋转指定角度,还是需要transforms.functional.rotate函数
官网地址: rotate — Torchvision 0.12 documentation
具体方法:pytorch数据增强之指定几个角度旋转 - 知乎
建议函数还是看官方文档更为靠谱。。。