"vscode:/vscode.git/clone" did not exist on "94b5edeb5384ea2a46533a11dd5938b2c859bf5c"
register.cpp 429 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#include "Video.h"

namespace {

static auto registerVideo =
    torch::class_<Video>("torchvision", "Video")
        .def(torch::init<std::string, std::string>())
        .def("get_current_stream", &Video::getCurrentStream)
        .def("set_current_stream", &Video::setCurrentStream)
        .def("get_metadata", &Video::getStreamMetadata)
        .def("seek", &Video::Seek)
        .def("next", &Video::Next);

} // namespace