|
|
@@ -1,6 +1,7 @@
|
|
|
#include <sstream>
|
|
|
|
|
|
#include "pybind11/pybind11.h"
|
|
|
+#include "pybind11/numpy.h"
|
|
|
#include "opencv2/opencv.hpp"
|
|
|
#include "infer.hpp"
|
|
|
#include "resnet.hpp"
|
|
|
@@ -45,6 +46,7 @@ namespace pybind11::detail {
|
|
|
}
|
|
|
|
|
|
class TrtResnetInfer{
|
|
|
+public:
|
|
|
TrtResnetInfer(std::string model_path)
|
|
|
{
|
|
|
instance_ = resnet::load(model_path);
|
