bugfix

src/infer/trt/yolo.cu

@@ -1,4 +1,4 @@
-#include "model/yolo.hpp"
+#include "infer/trt/yolo.hpp"
 #include <vector>
 #include <memory>
 #include "infer/slice/slice.hpp"

src/main.cpp

@@ -22,7 +22,7 @@ int main()
         "refrigerator", "book", "clock", "vase", "scissors", "teddy bear", "hair drier",
         "toothbrush"
     }; 
-    std::shared_ptr<Infer> model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.5, 0.45);
+    std::shared_ptr<Infer> model = load("model/yolov8.engine", ModelType::YOLO11, names, 0, 0.5, 0.45);
 
     std::shared_ptr<Node::StreamNode> src_node0   = std::make_shared<Node::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/101");
     src_node0->set_skip_frame(10);