Skip to main content

Crate burn_vision

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
  • nms (Non-Maximum Suppression)

Re-exports§

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

Modules§

backends
Backend implementations for JIT and CPU
utils
Module for vision/image utilities

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().
NmsOptions
Non-Maximum Suppression options.
Point
2D Point used for vision ops. Coordinates start at the top left.
Size
2D size used for vision ops.
Transform2D
2D point transformation

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
Nms
Non-maximum suppression tensor operations
QVisionOps
Vision ops on quantized float tensors
VisionBackend
Vision capable backend, implemented by each backend