File size: 1,273 Bytes
d7e58f0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
from .compose import Compose
from .formatting import (
    Collect,
    ImageToTensor,
    ToNumpy,
    ToPIL,
    ToTensor,
    Transpose,
    to_tensor,
)
from .hybrik_transforms import (
    GenerateHybrIKTarget,
    HybrIKAffine,
    HybrIKRandomFlip,
    NewKeypointsSelection,
    RandomDPG,
    RandomOcclusion,
)
from .loading import LoadImageFromFile
from .synthetic_occlusion_augmentation import SyntheticOcclusion
from .transforms import (
    BBoxCenterJitter,
    CenterCrop,
    ColorJitter,
    GetRandomScaleRotation,
    Lighting,
    MeshAffine,
    MeshAffineED,
    Normalize,
    RandomChannelNoise,
    RandomHorizontalFlip,
    Rotation,
    SimulateLowRes,
)

__all__ = [
    'Compose',
    'to_tensor',
    'ToTensor',
    'ImageToTensor',
    'ToPIL',
    'ToNumpy',
    'Transpose',
    'Collect',
    'LoadImageFromFile',
    'CenterCrop',
    'RandomHorizontalFlip',
    'ColorJitter',
    'Lighting',
    'RandomChannelNoise',
    'GetRandomScaleRotation',
    'MeshAffine',
    'MeshAffineED',
    'HybrIKRandomFlip',
    'HybrIKAffine',
    'GenerateHybrIKTarget',
    'RandomDPG',
    'RandomOcclusion',
    'Rotation',
    'NewKeypointsSelection',
    'Normalize',
    'SyntheticOcclusion',
    'BBoxCenterJitter',
    'SimulateLowRes',
]