Albumentations库中RandomRotate90增强功能的图像应用扩展

在计算机视觉领域，数据增强是提升模型泛化能力的重要手段。Albumentations作为一款流行的图像增强库，其RandomRotate90变换功能最近得到了重要扩展，增加了对多图像同时旋转的支持。

功能背景

RandomRotate90是Albumentations中常用的几何变换之一，它能够以90度的倍数随机旋转输入图像。在之前的版本中，该变换主要针对单张图像进行处理。然而，在实际应用中，我们经常需要同时对多张相关图像（如RGB图像和对应的掩码图像）进行完全一致的旋转操作，以保持数据间的一致性。

技术实现

新版本通过扩展底层功能函数，增加了对axis参数的支持，使其能够与NumPy的rot90函数保持兼容。具体实现上：

1. 修改了基础旋转函数，使其能够接受axis参数
2. 针对单图像和多图像场景分别实现：
   • apply_to_image：处理单张图像
   • apply_to_images：处理多张图像序列

这种设计保持了API的一致性，同时提供了更灵活的功能。在内部实现上，两者主要区别在于传递给旋转函数的axis参数不同。

应用价值

这一改进为以下场景提供了便利：

1. 多模态数据处理：同时旋转RGB图像和深度图
2. 语义分割任务：保持原始图像和标注掩码的空间对应关系
3. 目标检测：确保图像旋转后边界框坐标的正确变换
4. 多视角系统：处理来自多个摄像头的同步图像数据

使用建议

开发者现在可以更高效地处理相关图像集的旋转增强，无需担心各图像间旋转不一致的问题。这一改进也体现了Albumentations库对实际应用场景需求的快速响应能力，进一步巩固了其作为计算机视觉领域重要工具的地位。

该功能的加入使得Albumentations在多图像协同增强方面更加完善，为复杂视觉任务的预处理流程提供了更强大的支持。