MATLAB

imshowpair

  1. figure,imshowpair(X0,Twarp,'falsecolor')

Python

https://github.com/lebedov/imshowpair

  1. #!/usr/bin/env python3
  3. # Copyright (c) 2018, Lev E. Givon
  4. # All rights reserved.
  5. # Distributed under the terms of the BSD license:
  6. # http://www.opensource.org/licenses/bsd-license
  8. import matplotlib.pyplot as plt
  9. import numpy as np
  11. def imshowpair(a, b, method=None, show_all=True, axes_visible=False, cmap=None,
  12.                interpolation=None, grid=False, *args, **kwargs):
  13.     """
  14.     Compare images.
  15.     Parameters
  16.     ----------
  17.     a, b : numpy.ndarray
  18.         Images to compare.
  19.     method : callable
  20.         Callable to apply to images. Must support at least two arguments;
  21.         additional specified arguments are passed to `method`.
  22.         If None, both images are displayed side by side.
  23.     show_all : bool
  24.         If True and `method` is defined, display original images
  25.         alongside `method(a, b)`.
  26.     axes_visible : bool
  27.         Show axes if True.
  28.     cmap : matplotlib.colors.Colormap
  29.         Color map to use when displaying images.
  30.     interpolation : str
  31.         Interpolation method for `imshow` to use.
  32.     grid : bool
  33.         If True, display grid.
  34.     """
  36.     def turn_off_ticks(ax):
  37.         for tic in ax.xaxis.get_major_ticks():
  38.             tic.tick1On = tic.tick2On = False
  39.         for tic in ax.yaxis.get_major_ticks():
  40.             tic.tick1On = tic.tick2On = False
  42.     a = np.asarray(a)
  43.     b = np.asarray(b)
  44.     a_shape = a.shape
  45.     b_shape = b.shape
  47.     ax_list = []
  48.     if method == None:
  49.         ax0 = ax = plt.subplot(121)
  50.         plt.imshow(a, origin='upper', cmap=cmap, interpolation=interpolation)
  51.         plt.grid(grid)
  52.         turn_off_ticks(ax)
  53.         ax_list.append(ax)
  54.         ax = plt.subplot(122, sharex=ax0, sharey=ax0)
  55.         plt.imshow(b, origin='upper', cmap=cmap, interpolation=interpolation)
  56.         plt.grid(grid)
  57.         turn_off_ticks(ax)
  58.         if a_shape == b_shape:
  59.             ax.set_yticklabels([])
  60.         ax_list.append(ax)
  61.         plt.subplots_adjust()
  62.     elif callable(method):
  63.         c = method(a, b, *args, **kwargs)
  64.         c_shape = c.shape
  65.         if show_all:
  66.             ax0 = ax = plt.subplot(131)
  67.             plt.imshow(a, origin='upper', cmap=cmap, interpolation=interpolation)
  68.             plt.grid(grid)
  69.             turn_off_ticks(ax)
  70.             ax_list.append(ax)
  71.             ax = plt.subplot(132, sharex=ax0, sharey=ax0)
  72.             plt.imshow(b, origin='upper', cmap=cmap, interpolation=interpolation)
  73.             plt.grid(grid)
  74.             turn_off_ticks(ax)
  75.             if a_shape == b_shape:
  76.                 ax.set_yticklabels([])
  77.             ax_list.append(ax)
  78.             ax = plt.subplot(133, sharex=ax0, sharey=ax0)
  79.             plt.imshow(c, origin='upper', cmap=cmap, interpolation=interpolation)
  80.             plt.grid(grid)
  81.             turn_off_ticks(ax)
  82.             plt.subplots_adjust()
  83.             if c_shape == b_shape:
  84.                 ax.set_yticklabels([])
  85.             ax_list.append(ax)
  86.         else:
  87.             ax = plt.subplot(111)
  88.             plt.imshow(c, origin='upper', cmap=cmap, interpolation=interpolation)
  89.             plt.grid(grid)           
  90.             turn_off_ticks(ax)
  91.             if c_shape == b_shape:
  92.                 ax.set_yticklabels([])
  93.             ax_list.append(ax)
  94.     else:
  95.         NotImplementedError
  96.     if not axes_visible:
  97.         for ax in ax_list:
  98.             ax.set_xticklabels([])
  99.             ax.set_yticklabels([])