mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 03:04:42 +08:00
emmanuelle
55f5103dd8
Modified travis_script.sh to account for the new structure of the gallery Added README.txt files in directories of gallery examples Fixed references to gallery images in user guide pages Fixed broken links
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""
|
|
============================
|
|
Block views on images/arrays
|
|
============================
|
|
|
|
This example illustrates the use of `view_as_blocks` from
|
|
`skimage.util.shape`. Block views can be incredibly useful when one
|
|
wants to perform local operations on non-overlapping image patches.
|
|
|
|
We use `astronaut` from `skimage.data` and virtually 'slice' it into square
|
|
blocks. Then, on each block, we either pool the mean, the max or the
|
|
median value of that block. The results are displayed altogether, along
|
|
with a spline interpolation of order 3 rescaling of the original `astronaut`
|
|
image.
|
|
|
|
"""
|
|
import numpy as np
|
|
from scipy import ndimage as ndi
|
|
from matplotlib import pyplot as plt
|
|
import matplotlib.cm as cm
|
|
|
|
from skimage import data
|
|
from skimage import color
|
|
from skimage.util.shape import view_as_blocks
|
|
|
|
|
|
# -- get `astronaut` from skimage.data in grayscale
|
|
l = color.rgb2gray(data.astronaut())
|
|
|
|
# -- size of blocks
|
|
block_shape = (4, 4)
|
|
|
|
# -- see `astronaut` as a matrix of blocks (of shape
|
|
# `block_shape`)
|
|
view = view_as_blocks(l, block_shape)
|
|
|
|
# -- collapse the last two dimensions in one
|
|
flatten_view = view.reshape(view.shape[0], view.shape[1], -1)
|
|
|
|
# -- resampling `astronaut` by taking either the `mean`,
|
|
# the `max` or the `median` value of each blocks.
|
|
mean_view = np.mean(flatten_view, axis=2)
|
|
max_view = np.max(flatten_view, axis=2)
|
|
median_view = np.median(flatten_view, axis=2)
|
|
|
|
# -- display resampled images
|
|
fig, axes = plt.subplots(2, 2, figsize=(8, 8), sharex=True, sharey=True)
|
|
ax0, ax1, ax2, ax3 = axes.ravel()
|
|
|
|
ax0.set_title("Original rescaled with\n spline interpolation (order=3)")
|
|
l_resized = ndi.zoom(l, 2, order=3)
|
|
#ax0.imshow(l_resized, cmap=cm.Greys_r)
|
|
ax0.imshow(l_resized, extent=(0, 128, 128, 0), interpolation='nearest', cmap=cm.Greys_r)
|
|
ax0.set_axis_off()
|
|
|
|
ax1.set_title("Block view with\n local mean pooling")
|
|
ax1.imshow(mean_view, interpolation='nearest', cmap=cm.Greys_r)
|
|
ax1.set_axis_off()
|
|
|
|
ax2.set_title("Block view with\n local max pooling")
|
|
ax2.imshow(max_view, interpolation='nearest', cmap=cm.Greys_r)
|
|
ax2.set_axis_off()
|
|
|
|
ax3.set_title("Block view with\n local median pooling")
|
|
ax3.imshow(median_view, interpolation='nearest', cmap=cm.Greys_r)
|
|
ax3.set_axis_off()
|
|
|
|
fig.subplots_adjust(hspace=0.4, wspace=0.4)
|
|
plt.show()
|