|
@@ -8,10 +8,7 @@
|
|
|
|
|
|
void test_depth()
|
|
void test_depth()
|
|
{
|
|
{
|
|
- // 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, GNode::DecodeType::GPU);
|
|
|
|
- // src_node0->set_skip_frame(1);
|
|
|
|
-
|
|
|
|
- std::shared_ptr<GNode::StreamNode> src_node0 = std::make_shared<GNode::StreamNode>("src0", "carperson.mp4", 0, GNode::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, GNode::DecodeType::GPU);
|
|
src_node0->set_skip_frame(1);
|
|
src_node0->set_skip_frame(1);
|
|
|
|
|
|
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);
|
|
@@ -35,7 +32,8 @@ void test_yolo()
|
|
{
|
|
{
|
|
// std::vector<std::string> names = { "person", "clothes", "vest" };
|
|
// std::vector<std::string> names = { "person", "clothes", "vest" };
|
|
std::vector<std::string> names = { "person", "car", "close", "open" };
|
|
std::vector<std::string> names = { "person", "car", "close", "open" };
|
|
- 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, GNode::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, GNode::DecodeType::GPU);
|
|
|
|
+ std::shared_ptr<GNode::StreamNode> src_node0 = std::make_shared<GNode::StreamNode>("src0", "carperson.mp4", 0, GNode::DecodeType::GPU);
|
|
src_node0->set_skip_frame(1);
|
|
src_node0->set_skip_frame(1);
|
|
|
|
|
|
std::shared_ptr<Infer> yolo_model = load("model/carperson.engine", ModelType::YOLOV5, names, 0, 0.25, 0.45);
|
|
std::shared_ptr<Infer> yolo_model = load("model/carperson.engine", ModelType::YOLOV5, names, 0, 0.25, 0.45);
|