edgefirst-image

High-performance image processing for edge AI inference pipelines.

This crate provides hardware-accelerated image loading, format conversion, resizing, rotation, and cropping operations optimized for ML preprocessing workflows.

Features

• Multiple backends — Automatic selection: G2D (if format supported) → CPU (same-size simple copies) → OpenGL (GPU) → CPU (general fallback)
• Format conversion - RGBA, RGB, NV12, NV16, YUYV, GREY, planar formats
• Geometric transforms - Resize, rotate (90° increments), flip, crop
• Zero-copy integration - Works with edgefirst-tensor DMA/SHM buffers
• JPEG/PNG support - Load and save with EXIF orientation handling

Quick Start

use edgefirst_image::{load_image, save_jpeg, ImageProcessor, ImageProcessorTrait, Rotation, Flip, Crop};
use edgefirst_tensor::{PixelFormat, DType, TensorDyn};

// Load an image
let bytes = std::fs::read("input.jpg")?;
let src = load_image(&bytes, Some(PixelFormat::Rgba), None)?;

// Create processor (auto-selects best backend)
let mut processor = ImageProcessor::new()?;

// Create destination with desired size
let mut dst = processor.create_image(640, 640, PixelFormat::Rgba, DType::U8, None)?;

// Convert with resize, rotation, letterboxing
processor.convert(
    &src,
    &mut dst,
    Rotation::None,
    Flip::None,
    Crop::letterbox(),  // Preserve aspect ratio
)?;

// Save result
save_jpeg(&dst, "output.jpg", 90)?;

Backends

Supported Formats

Note: Int8 variants (e.g. packed RGB int8, planar RGB int8) use DType::I8 with the corresponding PixelFormat rather than separate format constants.

Feature Flags

• opengl (default) - Enable OpenGL backend on Linux
• decoder (default) - Enable detection box rendering

Environment Variables

• EDGEFIRST_DISABLE_G2D - Disable G2D backend
• EDGEFIRST_DISABLE_GL - Disable OpenGL backend
• EDGEFIRST_DISABLE_CPU - Disable CPU backend
• EDGEFIRST_FORCE_BACKEND — Force a single backend: cpu, g2d, or opengl. Disables fallback chain.
• EDGEFIRST_FORCE_TRANSFER — Force GPU transfer method: pbo or dmabuf
• EDGEFIRST_TENSOR_FORCE_MEM — Set to 1 to force heap memory (disables DMA/SHM)

Segmentation Mask Rendering

Three rendering pipelines for YOLO instance segmentation masks:

Fused GPU Proto Path (draw_masks_proto)

Computes sigmoid(coefficients @ protos) per-pixel in a fragment shader — no intermediate mask materialization. Preferred for real-time overlay.

let (detections, proto_data) = decoder.decode_quantized_proto(&outputs)?;
processor.draw_masks_proto(&mut frame, &detections, &proto_data)?;

Hybrid CPU+GPU Path

CPU materializes binary masks (materialize_segmentations()), then OpenGL overlays them. Auto-selected when both CPU and GL backends are available.

Atlas Decode Path (decode_masks_atlas)

Renders all detection masks into a compact vertical strip atlas via GPU, reads back as uint8 arrays. Use when you need per-instance mask pixels for downstream processing.

let masks = processor.decode_masks_atlas(&detections, &proto_data, 640, 640)?;

Shader Variants

Int8 Interpolation Mode

Control quantized proto interpolation quality:

processor.set_int8_interpolation_mode(Int8InterpolationMode::Bilinear);

See BENCHMARKS.md for per-platform performance numbers.

Zero-Copy Model Input

Use create_image() to allocate the destination tensor with the processor's optimal memory backend (DMA-buf, PBO, or system memory). This enables zero-copy GPU paths that direct Tensor::new() allocation cannot achieve:

let mut dst = processor.create_image(640, 640, PixelFormat::Rgb, DType::U8, None)?;
processor.convert(&src, &mut dst, Rotation::None, Flip::None, Crop::letterbox())?;

If you need to write into a pre-allocated buffer with a specific memory type (e.g. an NPU-bound tensor), you can still use direct allocation:

let mut model_input = Tensor::<u8>::new(&[640, 640, 3], None, None)?;
model_input.set_format(PixelFormat::Rgb)?;
let mut dst = TensorDyn::from(model_input);
processor.convert(&src, &mut dst, Rotation::None, Flip::None, Crop::letterbox())?;

Multiplane NV12/NV16

For V4L2 multi-planar DMA-BUF buffers (separate Y and UV file descriptors):

let img = Tensor::from_planes(y_tensor, uv_tensor, PixelFormat::Nv12)?;
let src = TensorDyn::from(img);
processor.convert(&src, &mut dst, Rotation::None, Flip::None, Crop::default())?;

The OpenGL backend imports each plane's DMA-BUF fd separately for zero-copy GPU access.

License

Licensed under the Apache License, Version 2.0. See LICENSE for details.