|
|
|
@@ -32,7 +32,7 @@ from .validators import check_prob, check_crop, check_resize_interpolation, chec |
|
|
|
check_normalize_py, check_random_crop, check_random_color_adjust, check_random_rotation, \ |
|
|
|
check_transforms_list, check_random_apply, check_ten_crop, check_num_channels, check_pad, \ |
|
|
|
check_random_perspective, check_random_erasing, check_cutout, check_linear_transform, check_random_affine, \ |
|
|
|
check_mix_up, check_positive_degrees, check_uniform_augment_py |
|
|
|
check_mix_up, check_positive_degrees, check_uniform_augment_py, check_compose_list |
|
|
|
from .utils import Inter, Border |
|
|
|
|
|
|
|
DE_PY_INTER_MODE = {Inter.NEAREST: Image.NEAREST, |
|
|
|
@@ -75,6 +75,7 @@ class ComposeOp: |
|
|
|
>>> dataset = dataset.map(input_columns="image", operations=transform()) |
|
|
|
""" |
|
|
|
|
|
|
|
@check_compose_list |
|
|
|
def __init__(self, transforms): |
|
|
|
self.transforms = transforms |
|
|
|
|
|
|
|
|