Python wrapper around TinyEXR and stb_image
Project description
Simple Image IO
A lightweight C# and Python wrapper to read and write RGB images from / to various file formats. Supports .exr (with layers) via tinyexr and a number of other formats (including .png, .jpg, and .bmp) via stb_image and stb_image_write. We also implement our own importer and exporter for PFM. In addition, the package offers some basic image manipulation functionality and error metrics.
The C# wrapper further offers utilities for thread-safe atomic splatting of pixel values, and sending image data to the tev viewer via sockets. It also contains a very basic wrapper around Intel Open Image Denoise.
The Nuget package contains prebuilt binaries of the C++ wrapper for x86-64 Windows, Ubuntu, and macOS (.github/workflows/build.yml). The Python package is set up to automatically download an adequate CMake version and compile the C++ code on any platform.
Usage example (C#)
The following creates a one pixel image and writes it to various file formats:
RgbImage img = new(width: 1, height: 1);
img.SetPixel(0, 0, new(0.1f, 0.4f, 0.9f));
img.WriteToFile("test.exr");
img.WriteToFile("test.png");
img.WriteToFile("test.jpg");
Reading an image from one of the supported formats is equally simple:
RgbImage img = new("test.exr");
Console.WriteLine(img.GetPixel(0, 0).Luminance);
The pixel coordinate (0,0) corresponds to the top left corner of the image. Coordinates outside the valid range are clamped automatically; no error is raised. The framework also offers a MonochromeImage
with a single channel per pixel. Further, the base class ImageBase
can be used directly for images with arbitrary channel count (RgbImage
and MonochromeImage
only add some convenience functions like directly returning an RgbColor
object).
As an added bonus, the C# wrapper can connect to the tev HDR viewer and directly display image data via sockets. The following example generates a monochrome image and sends it to tev:
TevIpc tevIpc = new(); // uses tev's default port on localhost
// Create the image and initialize a tev sync
MonochromeImage image = new(width: 20, height: 10);
tevIpc.CreateImageSync("MyAwesomeImage", 20, 10, ("default", image));
// Pretend we are a renderer and write some image data.
image.SetPixel(0, 0, val: 1);
image.SetPixel(10, 0, val: 2);
image.SetPixel(0, 9, val: 5);
image.SetPixel(10, 9, val: 10);
// Tell the TevIpc class to update the image displayed by tev
// (this currently retransmitts all pixel values)
tevIpc.UpdateImage("MyAwesomeImage");
Usage example (Python)
The following creates a one pixel image, writes it to various file formats, reads one of them back in, and prints the red color channel of the pixel:
import simpleimageio as sio
sio.write("test.exr", [[[0.1, 0.4, 0.9]]])
sio.write("test.png", [[[0.1, 0.4, 0.9]]])
sio.write("test.jpg", [[[0.1, 0.4, 0.9]]])
img = sio.read("test.exr")
print(img[0,0,0])
In Python, an image is a 3D row-major array, where [0,0,0]
is the red color channel of the top left corner.
The convention is compatible with most other libraries that make use of numpy arrays for image represenation, like matplotlib.
Building from source
If you are on an architecture different from x86-64, you will need to compile the C++ wrapper from source. Below, you can find instructions on how to accomplish that.
Dependencies
All dependencies are header-only and included in the repository. Building requires
- a C++11 (or newer) compiler
- CMake
- .NET 5.0 (or newer)
- Python ≥ 3.6
Building the C# wrapper on x86-64 Windows, Linux, or macOS
Build the C++ low level library with CMake:
mkdir build
cd build
cmake .. -DCMAKE_BUILD_TYPE=Release
cmake --build . --config Release
cd ..
Compile and run the tests (optional):
dotnet test
That's it. Simply add a reference to SimpleImageIO/SimpleImageIO.csproj
to your project and you should be up and running.
MacOS and Open Image Denoise
The official Open Image Denoise binaries have the rpath set to @executable_path/
. However, the libtbb.12.dylib
file will be next to the libOpenImageDenoise.1.dylib
and we do not know where that is relative to our executable. Therefore, we need to add @loader_path/
to the rpath:
install_name_tool -add_rpath @loader_path/ libOpenImageDenoise.1.dylib
Building the C# wrapper on other platforms
The SimpleImageIO.csproj file needs to copy the correct .dll / .so / .dylib file to the appropriate runtime folder. Currently, the runtime identifiers (RID) and copy instructions are only set for the x86-64 versions of Windows, Linux, and macOS. To run the framework on other architectures, you will need to add them to the .csproj file. You can find the right RID for your platform here: https://docs.microsoft.com/en-us/dotnet/core/rid-catalog.
Note that, currently, Open Image Denoise is included in binary from. The Denoiser
class can therefore not be used on platforms other than x86-64 Windows, Linux, or macOS. Attempting to use it on other platforms will cause a DllNotFound
exception.
Then, you should be able to follow the steps above and proceed as usual.
Building the Python wrapper
Simply running:
python -m build
will automatically fetch an adequate version of CMake, compile the shared library, and build the Python package. You can then simply install the result via:
pip install ./dist/SimpleImageIO-*.whl
The tests can be run via:
cd PyTest
python -m unittest
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.