File size: 912 Bytes
cf84204 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """Augmentation strategies for GraPHFormer."""
from .tree_augmentations import (
Compose,
RandomDropSubTrees, RandomSkipParentNode, RandomSwapSiblingSubTrees,
RandomRotate, RandomJitter, RandomShift, RandomFlip,
RandomScaleCoords, RandomScaleFeats, RandomMaskFeats,
RandomElasticate, RandomJitterLength,
)
from .persistence_augmentations import (
PersistenceSpaceAugmentation, SigmaVariationAugmentation,
CombinedPersistenceAugmentation, get_default_augmentation,
)
__all__ = [
"Compose",
"RandomDropSubTrees", "RandomSkipParentNode", "RandomSwapSiblingSubTrees",
"RandomRotate", "RandomJitter", "RandomShift", "RandomFlip",
"RandomScaleCoords", "RandomScaleFeats", "RandomMaskFeats",
"RandomElasticate", "RandomJitterLength",
"PersistenceSpaceAugmentation", "SigmaVariationAugmentation",
"CombinedPersistenceAugmentation", "get_default_augmentation",
]
|