Skip to content

Commit 8f6481b

Browse files
authored
Update prepro.py
1 parent 04773c2 commit 8f6481b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/prepro.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1746,7 +1746,7 @@ def illumination(x, gamma = 1, contrast=1, saturation=1, is_random=False):
17461746
gamma = np.random.uniform(1.0, 5.0)
17471747
else:
17481748
gamma = 1
1749-
im_ = tl.prepro.brightness(x, gamma=gamma, gain=1, is_random=False)
1749+
im_ = brightness(x, gamma=gamma, gain=1, is_random=False)
17501750

17511751
# print("using contrast and saturation")
17521752
image = Image.fromarray(im_) # array -> PIL
@@ -1757,7 +1757,7 @@ def illumination(x, gamma = 1, contrast=1, saturation=1, is_random=False):
17571757
image = saturation_adjust.enhance(np.random.uniform(0.7,1.0))
17581758
im_ = np.array(image) # PIL -> array
17591759
else:
1760-
im_ = tl.prepro.brightness(im_, gamma=gamma, gain=1, is_random=False)
1760+
im_ = brightness(im_, gamma=gamma, gain=1, is_random=False)
17611761
image = Image.fromarray(im_) # array -> PIL
17621762
contrast_adjust = ImageEnhance.Contrast(image)
17631763
image = contrast_adjust.enhance(contrast)

0 commit comments

Comments
 (0)