nvidia.dali.fn.segmentation#

The following table lists all operations available in nvidia.dali.fn.segmentation module:

Function

Device support

Short description

segmentation.random_mask_pixel

CPU

Selects random pixel coordinates in a mask, sampled from a uniform distribution.

segmentation.random_object_bbox

CPU

Randomly selects an object from a mask and returns its bounding box.

segmentation.select_masks

CPU

Selects a subset of polygons by their mask ids.