mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-02 12:31:56 +08:00
Extend plugins with imshow_collection
A first implementation is made for Matplotlib.
This commit is contained in:
@@ -147,6 +147,7 @@ def imshow(im, ax=None, show_cbar=None, **kwargs):
|
||||
if kwargs.get('cmap', None) == 'viridis':
|
||||
kwargs['cmap'] = viridis
|
||||
lo, hi, cmap = _get_display_range(im)
|
||||
|
||||
kwargs.setdefault('interpolation', 'nearest')
|
||||
kwargs.setdefault('cmap', cmap)
|
||||
kwargs.setdefault('vmin', lo)
|
||||
@@ -160,8 +161,20 @@ def imshow(im, ax=None, show_cbar=None, **kwargs):
|
||||
plt.colorbar(ax_im, cax=cax)
|
||||
ax.set_adjustable('box-forced')
|
||||
ax.get_figure().tight_layout()
|
||||
|
||||
return ax_im
|
||||
|
||||
|
||||
def imshow_collection(ic, *args, **kwargs):
|
||||
"""Display all images in the collection.
|
||||
|
||||
"""
|
||||
fig, axes = plt.subplots(1, len(ic))
|
||||
for n, image in enumerate(ic):
|
||||
kwargs['axis'] = axes[n]
|
||||
imshow(image, *args, **kwargs)
|
||||
|
||||
|
||||
imread = plt.imread
|
||||
show = plt.show
|
||||
|
||||
|
||||
Reference in New Issue
Block a user