Perform gamma correction on an image.
| 19 | |
| 20 | |
| 21 | class AdjustGamma(object): |
| 22 | """Perform gamma correction on an image.""" |
| 23 | |
| 24 | def __init__(self, gamma, gain=1): |
| 25 | self.gamma = gamma |
| 26 | self.gain = gain |
| 27 | |
| 28 | def __call__(self, img): |
| 29 | return TF.adjust_gamma(img, self.gamma, self.gain) |
| 30 | |
| 31 | |
| 32 | class TwoCropsTransform: |
nothing calls this directly
no outgoing calls
no test coverage detected