Function ExportOpLabel

python/mod_cvcuda/operators/OpLabel.cpp:142–232

} // namespace

void ExportOpLabel(py::module &m)
{
    using namespace pybind11::literals;

    py::enum_<NVCVLabelMaskType>(m, "LabelMaskType", py::arithmetic())
        .value("REMOVE_ISLANDS_OUTSIDE_MASK_ONLY", NVCV_REMOVE_ISLANDS_OUTSIDE_MASK_ONLY)
        .export_values();

    m.def("label", &Label, "src"_a, "connectivity"_a = NVCV_CONNECTIVITY_4_2D, "assign_labels"_a = NVCV_LABEL_FAST,
          "mask_type"_a = NVCV_REMOVE_ISLANDS_OUTSIDE_MASK_ONLY, py::kw_only(), "count"_a = false, "stats"_a = false,
          "max_labels"_a = 10000, "bg_label"_a = nullptr, "min_thresh"_a = nullptr, "max_thresh"_a = nullptr,
          "min_size"_a = nullptr, "mask"_a = nullptr, "stream"_a = nullptr, R"pbdoc(

        Executes the Label operation on the given cuda stream.

        See also:
            Refer to the CV-CUDA C API reference for the Label operator for more details and usage examples.

        Args:
            src (cvcuda.Tensor): Input tensor to label connected-component regions.
            connectivity (cvcuda.ConnectivityType, optional): Choice to control connectivity of input elements,
                                                              default is cvcuda.CONNECTIVITY_4_2D.
            assign_labels (cvcuda.LABEL, optional): Choice on how labels are assigned,
                                                    default is cvcuda.LABEL.FAST.
            mask_type (cvcuda.LabelMaskType, optional): Choice on how the mask is used,
                                                        default is cvcuda.REMOVE_ISLANDS_OUTSIDE_MASK_ONLY.
            count (bool, optional): Use True to return the count of valid labeled regions.
            stats (bool, optional): Use True to return the statistics of valid labeled regions.
            max_labels (Number, optional): Maximum number of labels to compute statistics for, default is 10000.
            bg_label (cvcuda.Tensor, optional): Background tensor to define input values to be considered background
                                         labels and thus ignored.
            min_thresh (cvcuda.Tensor, optional): Minimum threshold tensor to mask input values below it to be 0, and others 1.
            max_thresh (cvcuda.Tensor, optional): Maximum threshold tensor to mask input values above it to be 0, and others 1.
            min_size (cvcuda.Tensor, optional): Minimum size tensor to remove islands, i.e. labeled regions with number of
                                         elements less than the minimum size.
            mask (cvcuda.Tensor, optional): Mask tensor, its behavior is controlled by \ref mask_type.  One choice is to
                                     control island removal in addition to \ref min_size, i.e. regions with at
                                     least one element inside the mask (non-zero values) are not removed in case
                                     mask_type is cvcuda.REMOVE_ISLANDS_OUTSIDE_MASK_ONLY.
            stream (cvcuda.Stream, optional): CUDA Stream on which to perform the operation.

        Returns:
            Tuple[cvcuda.Tensor, cvcuda.Tensor, cvcuda.Tensor]: A tuple with output labels, count of regions and their statistics.
                                           The count or stats tensors may be None if theirs arguments are False.

        Caution:
            Restrictions to several arguments may apply. Check the C API references of the CV-CUDA operator.
    )pbdoc");

    m.def("label_into", &LabelInto, "dst"_a, "count"_a = nullptr, "stats"_a = nullptr, "src"_a,
          "connectivity"_a = NVCV_CONNECTIVITY_4_2D, "assign_labels"_a = NVCV_LABEL_FAST,
          "mask_type"_a = NVCV_REMOVE_ISLANDS_OUTSIDE_MASK_ONLY, py::kw_only(), "bg_label"_a = nullptr,
          "min_thresh"_a = nullptr, "max_thresh"_a = nullptr, "min_size"_a = nullptr, "mask"_a = nullptr,
          "stream"_a = nullptr, R"pbdoc(

        Executes the Label operation on the given cuda stream.

        See also: