对图像分类任务的数据进行操作。可以利用Compose类将图像预处理/增强操作进行组合。
paddlex.cls.transforms.Compose(transforms)
根据数据预处理/增强算子对输入数据进行操作。 使用示例
参数
- transforms (list): 数据预处理/数据增强列表。
paddlex.cls.transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225], min_val=[0, 0, 0], max_val=[255.0, 255.0, 255.0])
对图像进行标准化。
1.像素值减去min_val 2.像素值除以(max_val-min_val), 归一化到区间 [0.0, 1.0]。 3.对图像进行减均值除以标准差操作。
paddlex.cls.transforms.ResizeByShort(short_size=256, max_size=-1)
根据图像的短边调整图像大小(resize)。
paddlex.cls.transforms.CenterCrop(crop_size=224)
以图像中心点扩散裁剪长宽为crop_size的正方形
paddlex.cls.transforms.RandomCrop(crop_size=224, lower_scale=0.08, lower_ratio=3. / 4, upper_ratio=4. / 3)
对图像进行随机剪裁,模型训练时的数据增强操作。
paddlex.cls.transforms.RandomHorizontalFlip(prob=0.5)
以一定的概率对图像进行随机水平翻转,模型训练时的数据增强操作。
paddlex.cls.transforms.RandomVerticalFlip(prob=0.5)
以一定的概率对图像进行随机垂直翻转,模型训练时的数据增强操作。
paddlex.cls.transforms.RandomRotate(rotate_range=30, prob=0.5)
以一定的概率对图像在[-rotate_range, rotaterange]角度范围内进行旋转,模型训练时的数据增强操作。
paddlex.cls.transforms.RandomDistort(brightness_range=0.9, brightness_prob=0.5, contrast_range=0.9, contrast_prob=0.5, saturation_range=0.9, saturation_prob=0.5, hue_range=18, hue_prob=0.5)
以一定的概率对图像进行随机像素内容变换,模型训练时的数据增强操作。
【注意】如果输入是uint8/uint16的RGB图像,该数据增强必须在数据增强Normalize之前使用。
brightness_range, 1+brightness_range]中随机取值作为明亮度缩放因子scale,按照公式image = image * scale调整图像明亮度。默认值为0.9。contrast_range, 1+contrast_range]中随机取值作为对比度缩放因子scale,按照公式image = image * scale + (image_mean + 0.5) * (1 - scale)调整图像对比度。默认为0.9。saturation_range, 1+saturation_range]中随机取值作为饱和度缩放因子scale,按照公式image = gray * (1 - scale) + image * scale,其中gray = R * 299/1000 + G * 587/1000+ B * 114/1000。默认为0.9。hue_range, hue_range]中随机取值作为色相角度调整差值delta,按照公式hue = hue + delta调整色相角度 。默认为18,取值范围[0, 360]。