| 168 | |
| 169 | |
| 170 | class RandomContrast(object): |
| 171 | def __init__(self, lower=0.5, upper=1.5): |
| 172 | self.lower = lower |
| 173 | self.upper = upper |
| 174 | assert self.upper >= self.lower, "contrast upper must be >= lower." |
| 175 | assert self.lower >= 0, "contrast lower must be non-negative." |
| 176 | |
| 177 | # expects float image |
| 178 | def __call__(self, image, boxes=None, labels=None): |
| 179 | if random.randint(2): |
| 180 | alpha = random.uniform(self.lower, self.upper) |
| 181 | image *= alpha |
| 182 | return image, boxes, labels |
| 183 | |
| 184 | |
| 185 | class RandomBrightness(object): |