diff --git a/skimage/io/_plugins/matplotlib_plugin.py b/skimage/io/_plugins/matplotlib_plugin.py index ec36ce04..4aa0b7f1 100644 --- a/skimage/io/_plugins/matplotlib_plugin.py +++ b/skimage/io/_plugins/matplotlib_plugin.py @@ -1,12 +1,27 @@ import matplotlib.pyplot as plt -def imshow(*args, **kwargs): +def imshow(im, *args, **kwargs): + """Show the input image and return the current axes. + + Parameters + ---------- + im : array, shape (M, N[, 3]) + The image to display. + + *args, **kwargs : positional and keyword arguments + These are passed directly to `matplotlib.pyplot.imshow`. + + Returns + ------- + ax : `matplotlib.pyplot.Axes` + The axes showing the image. + """ if plt.gca().has_data(): plt.figure() kwargs.setdefault('interpolation', 'nearest') kwargs.setdefault('cmap', 'gray') - plt.imshow(*args, **kwargs) + return plt.imshow(im, *args, **kwargs) imread = plt.imread show = plt.show