leon 1 mese fa
parent
commit
8315ce8a27
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      src/main.cpp

+ 1 - 1
src/main.cpp

@@ -20,7 +20,7 @@ int main()
 
     std::shared_ptr<Infer> model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
     std::shared_ptr<Node::InferNode> infer_node1   = std::make_shared<Node::InferNode>("YOLO11");
-    infer_node->set_model_instance(model, ModelType::YOLO11);
+    infer_node1->set_model_instance(model, ModelType::YOLO11);
 
     std::shared_ptr<Infer> depth_model = load("model/depth.engine", ModelType::DEPTH_ANYTHING, {}, 0, 0.25, 0.45);
     std::shared_ptr<Node::InferNode> infer_node2   = std::make_shared<Node::InferNode>("DEPTH_ANYTHING");