INNER CODE UNIT · Python

get_composed_augmentations

meetps/pytorch-semseg · ptsemseg/augmentations/__init__.py:40

def get_composed_augmentations(aug_dict):
    if aug_dict is None:
        logger.info("Using No Augmentations")
        return None

    augmentations = []
    for aug_key, aug_param in aug_dict.items():
        augmentations.append(key2aug[aug_key](aug_param))
        logger.info("Using {} aug with params {}".format(aug_key, aug_param))
    return Compose(augmentations)

View source record →