Apply random enhancement for Color,Contrast,Brightness,Sharpness. Args: scale(float): enhancement degree is from [1-scale, 1+scale]
(img, scale)
| 170 | |
| 171 | |
| 172 | def enhance(img, scale): |
| 173 | '''Apply random enhancement for Color,Contrast,Brightness,Sharpness. |
| 174 | |
| 175 | Args: |
| 176 | scale(float): enhancement degree is from [1-scale, 1+scale] |
| 177 | ''' |
| 178 | enhance_value = [1.0, 1.0, 1.0, 1.0] |
| 179 | for i in range(4): |
| 180 | r = random.randint(0, 1) |
| 181 | if r: |
| 182 | enhance_value[i] = random.uniform(1 - scale, 1 + scale) |
| 183 | if not enhance_value[0] == 1.0: |
| 184 | enhancer = ImageEnhance.Color(img) |
| 185 | img = enhancer.enhance(enhance_value[0]) |
| 186 | if not enhance_value[1] == 1.0: |
| 187 | enhancer = ImageEnhance.Contrast(img) |
| 188 | img = enhancer.enhance(enhance_value[1]) |
| 189 | if not enhance_value[2] == 1.0: |
| 190 | enhancer = ImageEnhance.Brightness(img) |
| 191 | img = enhancer.enhance(enhance_value[2]) |
| 192 | if not enhance_value[3] == 1.0: |
| 193 | enhancer = ImageEnhance.Sharpness(img) |
| 194 | img = enhancer.enhance(enhance_value[3]) |
| 195 | return img |
| 196 | |
| 197 | |
| 198 | def flip(img): |