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=mask)
transformed_ima