Function get_transform

accessory/data/transform.py:68–77

(transform_type: str, size=224)

def get_transform(transform_type: str, size=224):
    if transform_type == "random_resized_crop":
        transform = T_random_resized_crop(size)
    elif transform_type == "resized_center_crop":
        transform = T_resized_center_crop(size)
    elif transform_type == "padded_resize":
        transform = T_padded_resize(size)
    else:
        raise ValueError("unknown transform type: transform_type")
    return transform