| 338 | |
| 339 | |
| 340 | class RandomMirror(object): |
| 341 | def __call__(self, image, boxes, classes): |
| 342 | _, width, _ = image.shape |
| 343 | if random.randint(2): |
| 344 | image = image[:, ::-1] |
| 345 | boxes = boxes.copy() |
| 346 | boxes[:, 0::2] = width - boxes[:, 2::-2] |
| 347 | return image, boxes, classes |
| 348 | |
| 349 | |
| 350 | class SwapChannels(object): |