Composes several augmentations together. Args: transforms (List[Transform]): list of transforms to compose. Example: >>> augmentations.Compose([ >>> transforms.CenterCrop(10), >>> transforms.ToTensor(), >>> ])
| 34 | |
| 35 | |
| 36 | class Compose(object): |
| 37 | """Composes several augmentations together. |
| 38 | Args: |
| 39 | transforms (List[Transform]): list of transforms to compose. |
| 40 | Example: |
| 41 | >>> augmentations.Compose([ |
| 42 | >>> transforms.CenterCrop(10), |
| 43 | >>> transforms.ToTensor(), |
| 44 | >>> ]) |
| 45 | """ |
| 46 | |
| 47 | def __init__(self, transforms): |
| 48 | self.transforms = transforms |
| 49 | |
| 50 | def __call__(self, img, boxes=None, labels=None): |
| 51 | for t in self.transforms: |
| 52 | img, boxes, labels = t(img, boxes, labels) |
| 53 | return img, boxes, labels |
| 54 | |
| 55 | |
| 56 | class Lambda(object): |