Skip to content
Snippets Groups Projects

Add bob_to_pillow and back conversion

Merged Amir MOHAMMADI requested to merge deprecation into master
2 files
+ 43
0
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 41
0
import numpy as np
from PIL import Image
def to_matplotlib(img):
@@ -45,6 +46,46 @@ def to_bob(img):
return np.moveaxis(img, -1, -3)
def bob_to_pillow(img):
"""Converts the floating or uint8 image to a Pillow Image.
Parameters
----------
img : numpy.ndarray
A gray-scale or RGB color image in Bob format (channels first)
Returns
-------
Image
An object of pillow.Image.
"""
# first convert to matplotlib format
img = to_matplotlib(img)
# if img is floating point, convert to uint8
if isinstance(img, np.floating):
# we expect float images to be between 0 and 1
img = np.clip(img, 0, 1) * 255
img = img.astype(np.uint8)
return Image.fromarray(img)
def pillow_to_bob(img):
"""Converts an RGB or gray-scale pillow image to Bob format
Parameters
----------
img : Image
A Pillow Image
Returns
-------
numpy.ndarray
Image in Bob format
"""
return to_bob(np.array(img))
def opencvbgr_to_bob(img):
"""Returns a view of the image from OpenCV BGR format to Bob RGB format.
This function works with images, batches of images, videos, and higher
Loading