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;
20
21#[cfg(feature = "export-tests")]
22#[allow(missing_docs)]
23pub mod tests;
24
25pub use base::*;
26pub use ops::*;
27pub use tensor::*;
28
29pub use backends::{KernelShape, create_structuring_element};