| |
| |
|
|
| 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() |
|
|
