Load an image into a numpy array with proper Exif orientation handling
Project description
image_to_numpy
Load an image file into a numpy array - while automatically rotating the image based on Exif orientation. Prevents upside-down and sideways images!
import image_to_numpy
img = image_to_numpy.load_image_file("my_file.jpg")
The image is automatically rotated into the correct orientation if the image contains Exif orientation metadata. Otherwise, it is loaded normally.
From there, you can pass the numpy array to any Python library that works with images in numpy array format, like face_recognition, Keras, etc.
Installation
You can install from PyPI:
pip install image_to_numpy
Usage
import image_to_numpy
img = image_to_numpy.load_image_file("my_file.jpg")
Your image is loaded - with the correct orientation!
By default, the image array is returned as a numpy array with 3-channels of 8-bit RGB data.
You can control the output format by passing in an optional mode
parameter:
import image_to_numpy
img = image_to_numpy.load_image_file("my_file.jpg", mode="RGB")
# Supported modes:
# 1 (1-bit pixels, black and white, stored with one pixel per byte)
# L (8-bit pixels, black and white)
# RGB (3x8-bit pixels, true color)
# RGBA (4x8-bit pixels, true color with transparency mask)
# CMYK (4x8-bit pixels, color separation)
# YCbCr (3x8-bit pixels, color video format)
# I (32-bit signed integer pixels)
# F (32-bit floating point pixels)
If you have matplotlib installed, here's a quick way to show your image on the screen:
import matplotlib.pyplot as plt
import image_to_numpy
img = image_to_numpy.load_image_file("my_file.jpg")
plt.imshow(img)
plt.show()
Thanks
- EXIF test images used in the unittests created by Dave Perrett / @daveperrett
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.