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