mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix random noise corruptor

Browse Source 
It was functioning as a color shift
This commit is contained in:
James Betker 2020-09-27 11:12:24 -06:00
parent c85da79697
commit d8c3fc9327
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/image_corruptor.py
View File

@ -78,7 +78,7 @@ class ImageCorruptor:
elif 'noise' in aug: elif 'noise' in aug:
# Block noise # Block noise
noise_intensity = (rand_int % 4 + 2) / 255.0 # Between 1-4 noise_intensity = (rand_int % 4 + 2) / 255.0 # Between 1-4
img += np.random.randn() * noise_intensity img += np.random.randn(*img.shape) * noise_intensity
elif 'jpeg' in aug: elif 'jpeg' in aug:
# JPEG compression # JPEG compression
qf = (rand_int % 20 + 5) # Between 5-25 qf = (rand_int % 20 + 5) # Between 5-25