目录

安装数据增强库：

指定旋转90度增强

图片在-180度到+180度之间任意旋转

---

安装数据增强库：

pip install albumentations

指定旋转90度增强

import albumentations as A# 在你的数据增强管道中加入下面的旋转增强
transform = A.Compose([...A.Rotate(limit=90, p=0.5),  # 有50%的概率旋转90度...
])

图片在-180度到+180度之间任意旋转

可以这样设置：

A.Rotate(limit=(-180, 180), p=1.0)

这里，limit=(-180, 180) 指定了旋转角度的范围从-180度到+180度（包含顺时针和逆时针旋转），p=1.0 表示对所有选定的图片应用旋转操作（100%的概率）。

通过调整 limit 参数，你可以灵活地控制旋转的角度范围，以适应不同的数据增强需求。这种方式非常适合于增加模型对不同旋转角度的鲁棒性，特别是在目标的方向变化对模型性能有较大影响的应用场景中。

不过，请记得旋转图片会影响到标注的位置和形状。albumentations 库会自动调整标注框以匹配旋转后的图片，但前提是你需要正确地将标注框信息（如边界框坐标）传递给增强管道。这确保了数据增强不会破坏原有的标注准确性，从而允许模型从增强的数据中有效学习。