first commit

nodes/infers/vp_trt_yolov8_classifier.h

@@ -0,0 +1,30 @@
+#pragma once
+
+#ifdef VP_WITH_TRT
+#include "../vp_secondary_infer_node.h"
+#include "../../third_party/trt_yolov8/trt_yolov8_classifier.h"
+
+namespace vp_nodes {
+    // universal yolov8 classifier based on tensorrt using third_party/trt_yolov8 library
+    class vp_trt_yolov8_classifier: public vp_secondary_infer_node
+    {
+    private:
+        /* data */
+        std::shared_ptr<trt_yolov8::trt_yolov8_classifier> yolov8_classifier = nullptr;
+    protected:
+        // we need a totally new logic for the whole infer combinations
+        // no separate step pre-defined needed in base class
+        virtual void run_infer_combinations(const std::vector<std::shared_ptr<vp_objects::vp_frame_meta>>& frame_meta_with_batch) override;
+        // override pure virtual method, for compile pass
+        virtual void postprocess(const std::vector<cv::Mat>& raw_outputs, const std::vector<std::shared_ptr<vp_objects::vp_frame_meta>>& frame_meta_with_batch) override;
+    public:
+        vp_trt_yolov8_classifier(std::string node_name, 
+                                std::string model_path,
+                                std::string labels_path = "",  
+                                std::vector<int> p_class_ids_applied_to = std::vector<int>(), 
+                                int min_width_applied_to = 0, 
+                                int min_height_applied_to = 0);
+        ~vp_trt_yolov8_classifier();
+    };
+}
+#endif