Module vision

Expand description

Vision module.

Modules§

dispatch_bool_dtype: Dispatches connected components based on the bool dtype, binding a concrete integer type to enable generic instantiations without extra trait bounds (after removing ElementComparison from Element).
dispatch_int_dtype: Dispatches connected components based on the int dtype, binding a concrete integer type to enable generic instantiations without extra trait bounds (after removing ElementComparison from Element).

BoxBlur: Blurs a batch of images with a uniform box filter, applied with a given probability. Built from a BoxBlurConfig.
BoxBlurConfig: Configuration for a BoxBlur augmentation.
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.
ConnectedStatsPrimitiveMetadata: Tensor metadata and ordinary fields for the corresponding backend extension value.
MedianBlur: Blurs a batch of images with a square median filter, applied with a given probability. Built from a MedianBlurConfig.
MedianBlurConfig: Configuration for a MedianBlur augmentation.
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

create_structuring_element: Create a structuring element tensor for use with morphology ops
filter2d: Depthwise convolution of a tensor with a 2D kernel.
median: Apply a depth-wise square median filter to the given images