File: utils.py

package info (click to toggle)
python-imgviz 1.7.5%2Bds-2
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 9,084 kB
  • sloc: python: 3,229; makefile: 15
file content (14 lines) | stat: -rw-r--r-- 303 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
 
import numpy as np
import PIL.Image


def pillow_to_numpy(img):
    img_numpy = np.asarray(img)
    if not img_numpy.flags.writeable:
        img_numpy = np.array(img)
    return img_numpy


def numpy_to_pillow(img, mode=None):
    img_pillow = PIL.Image.fromarray(img, mode=mode)
    return img_pillow