diff options
Diffstat (limited to 'inversion/visualize.py')
| -rw-r--r-- | inversion/visualize.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/inversion/visualize.py b/inversion/visualize.py index d17fe13..2461458 100644 --- a/inversion/visualize.py +++ b/inversion/visualize.py @@ -36,6 +36,9 @@ def data2img(data): rescaled = np.clip(rescaled, 0, 255) return np.rint(rescaled).astype('uint8') +def data2pil(data); + return Image.fromarray(data2img(data), mode='RGB') + def interleave(a, b): res = np.empty([a.shape[0] + b.shape[0]] + list(a.shape[1:]), dtype=a.dtype) res[0::2] = a |