Skip to content

Commit

Permalink
Merge pull request #454 from aleju/color_temperature
Browse files Browse the repository at this point in the history
Color temperature augmentation
  • Loading branch information
aleju authored Oct 16, 2019
2 parents 2d3fdb7 + 0a18185 commit 9e71c97
Show file tree
Hide file tree
Showing 4 changed files with 826 additions and 0 deletions.
5 changes: 5 additions & 0 deletions changelogs/master/added/20191013_change_color_temperature.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
# New Augmenter ChangeColorTemperature #454

* Added augmenter `imgaug.augmenters.color.ChangeColorTemperature`.
* Added function `imgaug.augmenters.color.change_color_temperatures_()`.
* Added function `imgaug.augmenters.color.change_color_temperature_()`.
17 changes: 17 additions & 0 deletions checks/check_color_temperature.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
from __future__ import print_function, division
import numpy as np
import imgaug as ia
import imgaug.augmenters as iaa


def main():
image = ia.quokka_square()
images_aug = []
for kelvin in np.linspace(1000, 10000, 64):
images_aug.append(iaa.ChangeColorTemperature(kelvin)(image=image))

ia.imshow(ia.draw_grid(images_aug))


if __name__ == "__main__":
main()
Loading

0 comments on commit 9e71c97

Please sign in to comment.