|
|
|
|
|
|
|
import numpy as np |
|
import matplotlib.pyplot as plt |
|
|
|
|
|
_figures = {} |
|
|
|
|
|
def show(im, waittime=0, id='plt', max_width=600): |
|
plt.ion() |
|
w = min(im.shape[1], max_width) |
|
h = max_width * (1.0 * im.shape[0]) / im.shape[1] if w == max_width else im.shape[0] |
|
plt.figure(id, figsize=(w / 80, h / 80), dpi=80) |
|
|
|
ax = plt.axes([0, 0, 1, 1], frameon=False) |
|
ax.get_xaxis().set_visible(False) |
|
ax.get_yaxis().set_visible(False) |
|
|
|
if np.issubdtype(im.dtype, np.floating): |
|
if np.max(im) > 1: |
|
factor = 255 / np.max(im) |
|
else: |
|
factor = 255 |
|
else: |
|
factor = 1 |
|
|
|
if np.atleast_3d(im).shape[2] == 3: |
|
data = np.uint8(im * factor)[:, :, ::-1] |
|
else: |
|
data = np.uint8(np.dstack((im, im, im)) * factor) |
|
|
|
if id in _figures and plt.fignum_exists(id): |
|
_figures[id].set_array(data) |
|
else: |
|
_figures[id] = plt.imshow(data) |
|
|
|
if waittime == 0: |
|
plt.waitforbuttonpress() |
|
else: |
|
plt.pause(waittime / 1000.) |
|
plt.ioff() |
|
|
