Slide 22
Slide 22 text
# extract patches
patches = extract_simple_patches_2d(img, patch_size)
# normalize patches
patches = patches.reshape(patches.shape[0], -1).astype(np.float64)
intercept = np.mean(patches, axis=0)
patches -= intercept
patches /= np.std(patches, axis=0)
# dictionary learning
model = MiniBatchDictionaryLearning(n_components=n_basis, alpha=1, n_iter=n_iter,
n_jobs=1)
model.fit(patches)
# reconstruction
reconstructed_patches = np.dot(code, model.components_)
reconstructed_patches = reconstructed_patches.reshape(len(patches), *patch_size)
reconstructed = reconstruct_from_simple_patches_2d(reconstructed_patches, img.shape)
Code: Image Reconstruction