Пример #1
0
 def denoise(self, var_in, var_out, niter, W):
     arr = np.array(self.vars[var_in], dtype=np.float32)
     img = tvdenoise(arr, niter=niter, W=W)
     self.vars[var_out] = img
Пример #2
0
if len(sys.argv) > 1:
    parser = argparse.ArgumentParser(description='Total-variation denoising')
    parser.add_argument('filename_in', metavar='in', help='the input file')
    parser.add_argument('-W', default=50.0, type=float,
        help='weight on regularization term')
    parser.add_argument('-n', default=10, type=int,
        help='number of iterations')
    args = parser.parse_args()
    
    filename = args.filename_in
    n = args.n
    W = args.W
else:
    filename = os.path.join(data_dir, 'lena256.tif')
    n = 10
    W = 50.0

im = imread(filename)
imshow(im)
im = np.array(im, dtype=np.float32)
im2 = tvdenoise(im, n, W)

def view_float(im):
    m = np.min(im)
    M = np.max(im)
    scale = float(M - m)
    rescaled = np.rint(255.0 * (im - m) / scale)
    imshow(np.array(rescaled, dtype=np.uint8))

view_float(im2)