NumPy array to PIL Image
(arr, mode)
| 61 | |
| 62 | |
| 63 | def array2image(arr, mode): |
| 64 | """NumPy array to PIL Image""" |
| 65 | arr = arr.swapaxes(1, 2).swapaxes(0, 2) |
| 66 | arr[arr < 0] = 0 |
| 67 | arr[arr > 255] = 255 |
| 68 | arr = numpy.fix(arr).astype(numpy.uint8) |
| 69 | return Image.frombytes(mode, arr.shape[1::-1], arr.tobytes()) |
| 70 | |
| 71 | |
| 72 | def load_image(path, mode=None, size=None): |