Spaces:
Runtime error
Runtime error
File size: 524 Bytes
7734d5b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
#include "cocoeval/cocoeval.h"
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("COCOevalAccumulate", &COCOeval::Accumulate, "COCOeval::Accumulate");
m.def(
"COCOevalEvaluateImages",
&COCOeval::EvaluateImages,
"COCOeval::EvaluateImages");
pybind11::class_<COCOeval::InstanceAnnotation>(m, "InstanceAnnotation")
.def(pybind11::init<uint64_t, double, double, bool, bool>());
pybind11::class_<COCOeval::ImageEvaluation>(m, "ImageEvaluation")
.def(pybind11::init<>());
}
|