| 120 | } // namespace |
| 121 | |
| 122 | void ExportOpCvtColor(py::module &m) |
| 123 | { |
| 124 | using namespace pybind11::literals; |
| 125 | py::options options; |
| 126 | options.disable_function_signatures(); |
| 127 | |
| 128 | m.def("cvtcolor", &CvtColor, "src"_a, "code"_a, py::kw_only(), "stream"_a = nullptr, R"pbdoc( |
| 129 | |
| 130 | cvcuda.cvtcolor(src: cvcuda.Tensor, code: cvcuda.ColorConversion, stream: Optional[cvcuda.Stream] = None) -> cvcuda.Tensor |
| 131 | |
| 132 | Executes the CVT Color operation on the given cuda stream. |
| 133 | |
| 134 | See also: |
| 135 | Refer to the CV-CUDA C API reference for the CVT Color operator |
| 136 | for more details and usage examples. |
| 137 | |
| 138 | Args: |
| 139 | src (cvcuda.Tensor): Input tensor containing one or more images. |
| 140 | code (cvcuda.ColorConversion): Code describing the desired color conversion. |
| 141 | stream (cvcuda.Stream, optional): CUDA Stream on which to perform the operation. |
| 142 | |
| 143 | Returns: |
| 144 | cvcuda.Tensor: The output tensor. |
| 145 | |
| 146 | Caution: |
| 147 | Restrictions to several arguments may apply. Check the C |
| 148 | API references of the CV-CUDA operator. |
| 149 | )pbdoc"); |
| 150 | |
| 151 | m.def("cvtcolor_into", &CvtColorInto, "dst"_a, "src"_a, "code"_a, py::kw_only(), "stream"_a = nullptr, R"pbdoc( |
| 152 | |
| 153 | cvcuda.cvtcolor_into(dst: cvcuda.Tensor, src: cvcuda.Tensor, code: cvcuda.ColorConversion, stream: Optional[cvcuda.Stream] = None) |
| 154 | |
| 155 | Executes the CVT Color operation on the given cuda stream. |
| 156 | |
| 157 | See also: |
| 158 | Refer to the CV-CUDA C API reference for the CVT Color operator |
| 159 | for more details and usage examples. |
| 160 | |
| 161 | Args: |
| 162 | dst (cvcuda.Tensor): Output tensor to store the result of the operation. |
| 163 | src (cvcuda.Tensor): Input tensor containing one or more images. |
| 164 | code (cvcuda.ColorConversion): Code describing the desired color conversion. |
| 165 | stream (cvcuda.Stream, optional): CUDA Stream on which to perform the operation. |
| 166 | |
| 167 | Returns: |
| 168 | None |
| 169 | |
| 170 | Caution: |
| 171 | Restrictions to several arguments may apply. Check the C |
| 172 | API references of the CV-CUDA operator. |
| 173 | )pbdoc"); |
| 174 | |
| 175 | m.def("cvtcolor", &CvtColorVarShape, "src"_a, "code"_a, py::kw_only(), "stream"_a = nullptr, R"pbdoc( |
| 176 | |
| 177 | cvcuda.cvtcolor(src: cvcuda.ImageBatchVarShape, code: cvcuda.ColorConversion, stream: Optional[cvcuda.Stream] = None) -> cvcuda.ImageBatchVarShape |
| 178 | |
| 179 | Executes the CVT Color operation on the given cuda stream. |