burn_vision/
lib.rs

1//! Vision ops for burn, with GPU acceleration where possible.
2//!
3//! # Operations
4//! Operation names are based on `opencv` wherever applicable.
5//!
6//! Currently implemented are:
7//! - `connected_components`
8//! - `connected_components_with_stats`
9//!
10
11#![warn(missing_docs)]
12
13extern crate alloc;
14
15/// Backend implementations for JIT and CPU
16pub mod backends;
17mod base;
18mod ops;
19mod tensor;
20mod transform;
21
22#[cfg(feature = "export-tests")]
23#[allow(missing_docs)]
24pub mod tests;
25
26pub use base::*;
27pub use ops::*;
28pub use tensor::*;
29pub use transform::*;
30
31/// Module for vision/image utilities
32pub mod utils;
33
34pub use backends::{KernelShape, create_structuring_element};