debug

src/interface.cpp

@@ -1,5 +1,5 @@
 #include <sstream>
-
+#include <iostream>
 #include "pybind11/pybind11.h"
 #include "pybind11/numpy.h"
 #include "opencv2/opencv.hpp"
@@ -59,6 +59,7 @@ public:
 
     resnet::Attribute forward(const cv::Mat& image)
     {
+        cout << image.size << std::endl;
         return instance_->forward(cvimg(image));
     }
 

test.py

@@ -6,4 +6,9 @@ infer = trtresnet.TrtResnetInfer("workspace/resnet.engine")
 # image = cv2.imread("workspace/inference/car.jpg")
 result = infer.forward_path("workspace/inference/car.jpg")
 
+
 print(result)
+
+image = cv2.imread("workspace/inference/car.jpg")
+
+infer.forward(image)