| 421 | |
| 422 | |
| 423 | class RandomRotate(object): |
| 424 | def __init__(self, degree): |
| 425 | self.degree = degree |
| 426 | |
| 427 | def __call__(self, img, mask): |
| 428 | rotate_degree = random.random() * 2 * self.degree - self.degree |
| 429 | return img.rotate(rotate_degree, Image.BICUBIC), mask.rotate( |
| 430 | rotate_degree, Image.NEAREST) |
| 431 | |
| 432 | |
| 433 | class RandomSizeAndCrop(object): |
nothing calls this directly
no outgoing calls
no test coverage detected