kornia-image 0.1.14

Image types and traits for generating and manipulating images
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

// Re-export the CpuAllocator from kornia-tensor
use kornia_tensor::allocator::TensorAllocatorError;
pub use kornia_tensor::CpuAllocator;
use kornia_tensor::TensorAllocator;

/// A marker trait for allocating and deallocating memory for images.
pub trait ImageAllocator: TensorAllocator {}

impl ImageAllocator for CpuAllocator {}

/// A no-op allocator for wrapping foreign-owned memory (numpy, CUDA, mmap, etc.)
/// as Rust Image objects without taking ownership.
#[derive(Clone)]
pub struct ForeignAllocator;

impl TensorAllocator for ForeignAllocator {
    fn alloc(&self, _layout: std::alloc::Layout) -> Result<*mut u8, TensorAllocatorError> {
        panic!("ForeignAllocator: alloc called on foreign-memory allocator");
    }

    fn dealloc(&self, _ptr: *mut u8, _layout: std::alloc::Layout) {}
}

impl ImageAllocator for ForeignAllocator {}