| 183 | |
| 184 | |
| 185 | class RandomBrightness(object): |
| 186 | def __init__(self, delta=32): |
| 187 | assert delta >= 0.0 |
| 188 | assert delta <= 255.0 |
| 189 | self.delta = delta |
| 190 | |
| 191 | def __call__(self, image, boxes=None, labels=None): |
| 192 | if random.randint(2): |
| 193 | delta = random.uniform(-self.delta, self.delta) |
| 194 | image += delta |
| 195 | return image, boxes, labels |
| 196 | |
| 197 | |
| 198 | class ToCV2Image(object): |