(self, image, masks, boxes, labels)
| 515 | self.rand_light_noise = RandomLightingNoise() |
| 516 | |
| 517 | def __call__(self, image, masks, boxes, labels): |
| 518 | im = image.copy() |
| 519 | im, masks, boxes, labels = self.rand_brightness(im, masks, boxes, labels) |
| 520 | if random.randint(2): |
| 521 | distort = Compose(self.pd[:-1]) |
| 522 | else: |
| 523 | distort = Compose(self.pd[1:]) |
| 524 | im, masks, boxes, labels = distort(im, masks, boxes, labels) |
| 525 | return self.rand_light_noise(im, masks, boxes, labels) |
| 526 | |
| 527 | class PrepareMasks(object): |
| 528 | """ |