Method Export

python/mod_cvcuda/nvcv/ImageBatch.cpp:205–232

}

void ImageBatchVarShape::Export(py::module &m)
{
    using namespace py::literals;

    py::class_<ImageBatchVarShape, std::shared_ptr<ImageBatchVarShape>, Container>(m, "ImageBatchVarShape",
                                                                                   "Batch of Images.")
        .def(py::init(&ImageBatchVarShape::Create), "capacity"_a,
             "Create a new ImageBatchVarShape object with the specified capacity.")
        .def_property_readonly("uniqueformat", &ImageBatchVarShape::uniqueFormat,
                               "Return True if all the images have the same format, False otherwise.")
        .def_property_readonly("maxsize", &ImageBatchVarShape::maxSize,
                               "Return the maximum size of the ImageBatchVarShape in bytes.")
        .def_property_readonly("capacity", &ImageBatchVarShape::capacity,
                               "Return the capacity of the ImageBatchVarShape in number of images.")
        .def("__len__", &ImageBatchVarShape::numImages, "Return the number of images in the ImageBatchVarShape.")
        .def(
            "__iter__", [](const ImageBatchVarShape &list) { return py::make_iterator(list); },
            "Return an iterator over the images in the ImageBatchVarShape.")
        .def("pushback", &ImageBatchVarShape::pushBack, "Add a new image to the end of the ImageBatchVarShape.")
        .def("pushback", &ImageBatchVarShape::pushBackMany, "Add multiple images to the end of the ImageBatchVarShape.")
        .def("popback", &ImageBatchVarShape::popBack, "count"_a = 1,
             "Remove one or more images from the end of the ImageBatchVarShape.")
        .def("clear", &ImageBatchVarShape::clear, "Remove all images from the ImageBatchVarShape.");

    m.def("as_images", &ImageBatchVarShape::WrapExternalBufferVector, py::arg_v("buffers", std::vector<py::object>{}),
          "format"_a = nvcv::FMT_NONE, py::keep_alive<0, 1>(),
          "Wrap a vector of external buffers as a batch of images, and tie the buffers lifetime to it");
}

} // namespace nvcvpy::priv