Albumentation
所有实现的变换。
变换以及支持的类型。此处
Classification
- 创建图片扩展流程
使用 Compose 类定义数据增强的流程。Compose 类接受一个包含所有图片变换的列表。
transform = A.Compose([
A.RandomCrop(width=256, height=256),
A.HorizontalFlip(p=0.5),
A.RandomBrightnessContrast(p=0.2),
])
- 读取图片
Albumentation采用 RGB 格式
- 传入图片,获取变换后图片
# 返回是字典
transformed_image = transform(image=image)["image"]
Segmentation
需要同时变换图片和 Mask 。1,2 两布同 Classification 相同,第三步
- 传入图片和 Mask
transformed = transform(image=image, mask=m

Albumentation是一个强大的计算机视觉库,提供多种数据增强技术,适用于分类、分割、检测和关键点任务。它支持RGB图像、Mask、bbox和不同格式的关键点,能确保在图像增强过程中保留重要信息。
最低0.47元/天 解锁文章
1179





