main.cpp 836 B

12345678910111213141516171819202122232425262728
  1. #include "nodes/base/base.hpp"
  2. #include "nodes/stream/streamNode.hpp"
  3. #include "nodes/infer/inferNode.hpp"
  4. #include "nodes/draw/drawNode.hpp"
  5. int main()
  6. {
  7. std::shared_ptr<Node::StreamNode> src_node = std::make_shared<Node::StreamNode>("src", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/101");
  8. src_node->set_skip_frame(10);
  9. std::shared_ptr<Node::InferNode> infer_node = std::make_shared<Node::InferNode>("infer");
  10. std::shared_ptr<Node::DrawNode> draw_node = std::make_shared<Node::DrawNode>("draw");
  11. Node::LinkNode(src_node, infer_node);
  12. Node::LinkNode(infer_node, draw_node);
  13. draw_node->start();
  14. infer_node->start();
  15. src_node->start();
  16. while(true)
  17. {
  18. std::this_thread::sleep_for(std::chrono::seconds(1));
  19. }
  20. return 0;
  21. }
  22. // TODO: 单模型多路复用