| 1234567891011121314151617181920212223 |
- from fastai.vision.image import TfmPixel
- import random
- #Contributed by Rani Horev. Thank you!
- def _noisify(x, pct_pixels_min:float=0.001, pct_pixels_max:float=0.4, noise_range:int=30):
- if noise_range > 255 or noise_range < 0:
- raise Exception('noise_range must be between 0 and 255, inclusively.')
- h,w = x.shape[1:]
- img_size = h * w
- mult = 10000.0
- pct_pixels = random.randrange(int(pct_pixels_min*mult), int(pct_pixels_max*mult))/mult
- noise_count = int(img_size * pct_pixels)
- for ii in range(noise_count):
- yy = random.randrange(h)
- xx = random.randrange(w)
- noise = random.randrange(-noise_range, noise_range)/255.0
- x[:,yy,xx].add_(noise)
- return x
- noisify = TfmPixel(_noisify)
|
