|
@@ -1,5 +1,7 @@
|
|
|
-#include "pybind11/pybind11.h"
|
|
|
|
|
|
|
+#include <sstream>
|
|
|
|
|
|
|
|
|
|
+#include "pybind11/pybind11.h"
|
|
|
|
|
+#include "opencv2/opencv.hpp"
|
|
|
#include "infer.hpp"
|
|
#include "infer.hpp"
|
|
|
#include "resnet.hpp"
|
|
#include "resnet.hpp"
|
|
|
#include "cpm.hpp"
|
|
#include "cpm.hpp"
|
|
@@ -71,7 +73,6 @@ private:
|
|
|
|
|
|
|
|
PYBIND11_MODULE(trtresnet, m){
|
|
PYBIND11_MODULE(trtresnet, m){
|
|
|
py::class_<resnet::Attribute>(m, "Attribute")
|
|
py::class_<resnet::Attribute>(m, "Attribute")
|
|
|
- .def(py::init<const std::string&, int>())
|
|
|
|
|
.def_readwrite("confidence", &resnet::Attribute::confidence)
|
|
.def_readwrite("confidence", &resnet::Attribute::confidence)
|
|
|
.def_readwrite("class_label", &resnet::Attribute::class_label)
|
|
.def_readwrite("class_label", &resnet::Attribute::class_label)
|
|
|
.def("__repr__", [](const resnet::Attribute &attr) {
|
|
.def("__repr__", [](const resnet::Attribute &attr) {
|
