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//! - `nms` (Non-Maximum Suppression)
10//!
11
12#![warn(missing_docs)]
13
14extern crate alloc;
15
16/// Backend implementations for JIT and CPU
17pub mod backends;
18mod base;
19mod ops;
20mod tensor;
21mod transform;
22
23pub use base::*;
24pub use ops::*;
25pub use tensor::*;
26pub use transform::*;
27
28/// Module for vision/image utilities
29pub mod utils;
30
31pub use backends::{KernelShape, create_structuring_element};