wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-25 13:30:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH: Massively reworked code to enable nD-support

Browse Source
This commit is contained in:
Egor Panfilov
2015-12-30 22:45:27 +03:00
parent f2f698b190
commit 89784631e7
3 changed files with 51 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/filters/plot_inpaint.py
+6 -4
View File
@@ -26,18 +26,20 @@ import matplotlib.pyplot as plt
from skimage import data, color
from skimage.restoration import inpaint
image_orig = color.rgb2gray(data.astronaut())
image_orig = data.astronaut()
# Create mask with three defect regions: left, middle, right respectively
mask = np.zeros_like(image_orig)
mask = np.zeros(image_orig.shape[:-1])
mask[20:60, 0:20] = 1
mask[200:300, 150:170] = 1
mask[50:100, 400:430] = 1
# Defect image over the same region in each color channel
image_defect = image_orig.copy()
image_defect[np.where(mask)] = 0
for layer in range(image_defect.shape[-1]):
image_defect[np.where(mask)] = 0
image_result = inpaint.inpaint_biharmonic(image_defect, mask)
image_result = inpaint.inpaint_biharmonic(image_defect, mask, multichannel=True)
fig, axes = plt.subplots(ncols=3, nrows=1)