14 lines
524 B
C++
14 lines
524 B
C++
|
#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<>());
|
||
|
|
}
|