|
@@ -8,19 +8,19 @@
|
|
|
|
|
|
void test_depth()
|
|
void test_depth()
|
|
{
|
|
{
|
|
- std::shared_ptr<Node::StreamNode> src_node0 = std::make_shared<Node::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, Node::DecodeType::GPU);
|
|
|
|
|
|
+ std::shared_ptr<GNode::StreamNode> src_node0 = std::make_shared<GNode::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, Node::DecodeType::GPU);
|
|
src_node0->set_skip_frame(10);
|
|
src_node0->set_skip_frame(10);
|
|
|
|
|
|
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_node = std::make_shared<Node::InferNode>("depth");
|
|
|
|
|
|
+ std::shared_ptr<GNode::InferNode> infer_node = std::make_shared<GNode::InferNode>("depth");
|
|
infer_node->set_model_instance(depth_model, ModelType::DEPTH_ANYTHING);
|
|
infer_node->set_model_instance(depth_model, ModelType::DEPTH_ANYTHING);
|
|
|
|
|
|
- std::shared_ptr<Node::DrawNode> draw_node = std::make_shared<Node::DrawNode>("draw");
|
|
|
|
- std::shared_ptr<Node::HttpPushNode> push_node = std::make_shared<Node::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
|
|
|
|
|
|
+ std::shared_ptr<GNode::DrawNode> draw_node = std::make_shared<GNode::DrawNode>("draw");
|
|
|
|
+ std::shared_ptr<GNode::HttpPushNode> push_node = std::make_shared<GNode::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
|
|
|
|
|
|
- Node::LinkNode(src_node0, infer_node);
|
|
|
|
- Node::LinkNode(infer_node, draw_node);
|
|
|
|
- Node::LinkNode(draw_node, push_node);
|
|
|
|
|
|
+ GNode::LinkNode(src_node0, infer_node);
|
|
|
|
+ GNode::LinkNode(infer_node, draw_node);
|
|
|
|
+ GNode::LinkNode(draw_node, push_node);
|
|
push_node->start();
|
|
push_node->start();
|
|
draw_node->start();
|
|
draw_node->start();
|
|
infer_node->start();
|
|
infer_node->start();
|
|
@@ -31,22 +31,22 @@ void test_depth()
|
|
void test_yolo()
|
|
void test_yolo()
|
|
{
|
|
{
|
|
std::vector<std::string> names = { "person", "clothes", "vest" };
|
|
std::vector<std::string> names = { "person", "clothes", "vest" };
|
|
- std::shared_ptr<Node::StreamNode> src_node0 = std::make_shared<Node::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, Node::DecodeType::GPU);
|
|
|
|
|
|
+ std::shared_ptr<GNode::StreamNode> src_node0 = std::make_shared<GNode::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, Node::DecodeType::GPU);
|
|
src_node0->set_skip_frame(10);
|
|
src_node0->set_skip_frame(10);
|
|
|
|
|
|
std::shared_ptr<Infer> yolo_model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
|
|
std::shared_ptr<Infer> yolo_model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
|
|
- std::shared_ptr<Node::InferNode> infer_node = std::make_shared<Node::InferNode>("yolo11");
|
|
|
|
|
|
+ std::shared_ptr<GNode::InferNode> infer_node = std::make_shared<GNode::InferNode>("yolo11");
|
|
infer_node->set_model_instance(yolo_model, ModelType::YOLO11);
|
|
infer_node->set_model_instance(yolo_model, ModelType::YOLO11);
|
|
|
|
|
|
- std::shared_ptr<Node::TrackNode> track_node = std::make_shared<Node::TrackNode>("tracker", 30, 30);
|
|
|
|
|
|
+ std::shared_ptr<GNode::TrackNode> track_node = std::make_shared<GNode::TrackNode>("tracker", 30, 30);
|
|
|
|
|
|
- std::shared_ptr<Node::DrawNode> draw_node = std::make_shared<Node::DrawNode>("draw");
|
|
|
|
- std::shared_ptr<Node::HttpPushNode> push_node = std::make_shared<Node::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
|
|
|
|
|
|
+ std::shared_ptr<GNode::DrawNode> draw_node = std::make_shared<GNode::DrawNode>("draw");
|
|
|
|
+ std::shared_ptr<GNode::HttpPushNode> push_node = std::make_shared<GNode::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
|
|
|
|
|
|
- Node::LinkNode(src_node0, infer_node);
|
|
|
|
- Node::LinkNode(infer_node, track_node);
|
|
|
|
- Node::LinkNode(track_node, draw_node);
|
|
|
|
- Node::LinkNode(draw_node, push_node);
|
|
|
|
|
|
+ GNode::LinkNode(src_node0, infer_node);
|
|
|
|
+ GNode::LinkNode(infer_node, track_node);
|
|
|
|
+ GNode::LinkNode(track_node, draw_node);
|
|
|
|
+ GNode::LinkNode(draw_node, push_node);
|
|
push_node->start();
|
|
push_node->start();
|
|
draw_node->start();
|
|
draw_node->start();
|
|
track_node->start();
|
|
track_node->start();
|
|
@@ -59,25 +59,25 @@ void test_multi()
|
|
{
|
|
{
|
|
std::vector<std::string> names = { "person", "clothes", "vest" };
|
|
std::vector<std::string> names = { "person", "clothes", "vest" };
|
|
|
|
|
|
- std::shared_ptr<Node::StreamNode> src_node0 = std::make_shared<Node::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/101", 0, Node::DecodeType::GPU);
|
|
|
|
|
|
+ std::shared_ptr<GNode::StreamNode> src_node0 = std::make_shared<GNode::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/101", 0, GNode::DecodeType::GPU);
|
|
src_node0->set_skip_frame(10);
|
|
src_node0->set_skip_frame(10);
|
|
|
|
|
|
- std::shared_ptr<Node::StreamNode> src_node1 = std::make_shared<Node::StreamNode>("src1", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, Node::DecodeType::GPU);
|
|
|
|
|
|
+ std::shared_ptr<GNode::StreamNode> src_node1 = std::make_shared<GNode::StreamNode>("src1", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, GNode::DecodeType::GPU);
|
|
src_node1->set_skip_frame(10);
|
|
src_node1->set_skip_frame(10);
|
|
|
|
|
|
- std::shared_ptr<Node::StreamNode> src_node2 = std::make_shared<Node::StreamNode>("src2", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/301", 0, Node::DecodeType::GPU);
|
|
|
|
|
|
+ std::shared_ptr<GNode::StreamNode> src_node2 = std::make_shared<GNode::StreamNode>("src2", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/301", 0, GNode::DecodeType::GPU);
|
|
src_node2->set_skip_frame(10);
|
|
src_node2->set_skip_frame(10);
|
|
|
|
|
|
std::shared_ptr<Infer> yolo_model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
|
|
std::shared_ptr<Infer> yolo_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<GNode::InferNode> infer_node1 = std::make_shared<GNode::InferNode>("yolo11");
|
|
infer_node1->set_model_instance(yolo_model, ModelType::YOLO11);
|
|
infer_node1->set_model_instance(yolo_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");
|
|
|
|
|
|
+ std::shared_ptr<GNode::InferNode> infer_node2 = std::make_shared<GNode::InferNode>("depth");
|
|
infer_node2->set_model_instance(depth_model, ModelType::DEPTH_ANYTHING);
|
|
infer_node2->set_model_instance(depth_model, ModelType::DEPTH_ANYTHING);
|
|
|
|
|
|
- std::shared_ptr<Node::DrawNode> draw_node = std::make_shared<Node::DrawNode>("draw");
|
|
|
|
- std::shared_ptr<Node::HttpPushNode> push_node = std::make_shared<Node::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
|
|
|
|
|
|
+ std::shared_ptr<GNode::DrawNode> draw_node = std::make_shared<GNode::DrawNode>("draw");
|
|
|
|
+ std::shared_ptr<GNode::HttpPushNode> push_node = std::make_shared<GNode::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
|
|
|
|
|
|
Node::LinkNode(src_node0, infer_node1);
|
|
Node::LinkNode(src_node0, infer_node1);
|
|
Node::LinkNode(src_node1, infer_node1);
|
|
Node::LinkNode(src_node1, infer_node1);
|