import skimage
from PIL import Image
import numpy as np
    

if __name__ == '__main__':
    origin = skimage.img_as_float(Image.open('./test_images/my_image_normal.bmp'))
    noisy = skimage.util.random_noise(origin, mode='gaussian', var=0.01)
    noisy = (noisy * 255).astype(np.uint8)
    skimage.io.imsave('./test_images/my_image_gaussiannoise.bmp', noisy)
    
    origin = skimage.img_as_float(Image.open('./test_images/my_image_normal.bmp'))
    noisy = skimage.util.random_noise(origin, mode='salt')
    noisy = (noisy * 255).astype(np.uint8)
    skimage.io.imsave('./test_images/my_image_saltnoise.bmp', noisy)