MCPcopy Create free account
hub / github.com/CVCUDA/CV-CUDA / ExportOpInpaint

Function ExportOpInpaint

python/mod_cvcuda/operators/OpInpaint.cpp:189–297  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

187} // namespace
188
189void ExportOpInpaint(py::module &m)
190{
191 using namespace pybind11::literals;
192 py::options options;
193 options.disable_function_signatures();
194
195 m.def("inpaint", &Inpaint, "src"_a, "masks"_a, "inpaintRadius"_a, py::kw_only(), "stream"_a = nullptr,
196 R"pbdoc(
197
198 cvcuda.inpaint(src: cvcuda.Tensor, masks: Tensor, inpaintRadius: float, stream: Optional[cvcuda.Stream] = None) -> cvcuda.Tensor
199
200 Executes the Inpaint operation on the given cuda stream.
201
202 See also:
203 Refer to the CV-CUDA C API reference for the Inpaint operator
204 for more details and usage examples.
205
206 Args:
207 src (cvcuda.Tensor): Input tensor containing one or more images.
208 masks (cvcuda.Tensor): Mask tensor, 8-bit 1-channel images. Non-zero pixels indicate the area that needs to be inpainted.
209 inpaintRadius (float): Radius of a circular neighborhood of each point inpainted that is considered by the algorithm.
210 stream (cvcuda.Stream, optional): CUDA Stream on which to perform the operation.
211
212 Returns:
213 cvcuda.Tensor: The output tensor.
214
215 Caution:
216 Restrictions to several arguments may apply. Check the C
217 API references of the CV-CUDA operator.
218 )pbdoc");
219
220 m.def("inpaint_into", &InpaintInto, "dst"_a, "src"_a, "masks"_a, "inpaintRadius"_a, py::kw_only(),
221 "stream"_a = nullptr, R"pbdoc(
222
223 cvcuda.inpaint_into(dst: cvcuda.Tensor, src: cvcuda.Tensor, masks: Tensor, inpaintRadius: float, stream: Optional[cvcuda.Stream] = None)
224
225 Executes the Inpaint operation on the given cuda stream.
226
227 See also:
228 Refer to the CV-CUDA C API reference for the Inpaint operator
229 for more details and usage examples.
230
231 Args:
232 dst (cvcuda.Tensor): Output tensor to store the result of the operation.
233 src (cvcuda.Tensor): Input tensor containing one or more images.
234 masks (cvcuda.Tensor): Mask tensor, 8-bit 1-channel images. Non-zero pixels indicate the area that needs to be inpainted.
235 inpaintRadius (float): Radius of a circular neighborhood of each point inpainted that is considered by the algorithm.
236 stream (cvcuda.Stream, optional): CUDA Stream on which to perform the operation.
237
238 Returns:
239 None
240
241 Caution:
242 Restrictions to several arguments may apply. Check the C
243 API references of the CV-CUDA operator.
244 )pbdoc");
245
246 m.def("inpaint", &InpaintVarShape, "src"_a, "masks"_a, "inpaintRadius"_a, py::kw_only(), "stream"_a = nullptr,

Callers 1

PYBIND11_MODULEFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected