import numpy as np from denoiser import Denoiser, disp #Create a 1500x1500 image from random numbers for demonstration #Try replacing this with your own image! img = np.random.rand(1500, 1500) #Initialize the denoising neural network noise_remover = Denoiser(visible_cuda="0") #Denoise a 512x512 crop from the image crop = img[:512, :512] denoised_crop = noise_remover.denoise_crop(crop) #Denoise the entire image denoised_img = noise_remover.denoise(img) disp(crop) #Crop before denoising disp(denoised_crop) #Crop after denoising disp(img) #Image before denoising disp(denoised_img) #Image after denoising