|
@@ -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<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");
|
|
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<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");
|
|
std::shared_ptr<Node::InferNode> infer_node2 = std::make_shared<Node::InferNode>("DEPTH_ANYTHING");
|