Crate burn_vision

Source
Expand description

Vision ops for burn, with GPU acceleration where possible.

§Operations

Operation names are based on opencv wherever applicable.

Currently implemented are:

  • connected_components
  • connected_components_with_stats

Re-exports§

pub use backends::KernelShape;
pub use backends::create_structuring_element;

Modules§

backends
Backend implementations for JIT and CPU
tests

Macros§

as_type
testgen_all
testgen_connected_components
testgen_morphology

Structs§

ConnectedStats
Stats collected by the connected components analysis
ConnectedStatsOptions
Which stats should be enabled for connected_components_with_stats. Currently only used by the GPU implementation to save on atomic operations for unneeded stats.
ConnectedStatsPrimitive
Primitive version of ConnectedStats, to be returned by the backend
MorphOptions
Options for morphology ops
MorphOptionsBuilder
Use builder syntax to set the inputs and finish with build().
Point
2D Point used for vision ops. Coordinates start at the top left.
Size
2D size used for vision ops.

Enums§

BorderType
Morphology border type
Connectivity
Connected components connectivity

Traits§

BoolVisionOps
Vision ops on bool tensors
ConnectedComponents
Connected components tensor extensions
FloatVisionOps
Vision ops on float tensors
IntVisionOps
Vision ops on int tensors
Morphology
Morphology tensor operations
MorphologyKind
Morphology tensor operations
QVisionOps
Vision ops on quantized float tensors
VisionBackend
Vision capable backend, implemented by each backend