leon 3 tygodni temu
rodzic
commit
e108364ae0

+ 1 - 1
src/nodes/stream/streamNode.cpp

@@ -121,7 +121,7 @@ void StreamNode::work()
             }
             else
             {
-                process_stream_foler();
+                process_stream_folder();
             }
 
             PLOGI.printf("StreamNode [%s]: Stream processing finished or stopped (Status: %d).",

+ 1 - 0
src/nodes/stream/streamNode.hpp

@@ -85,6 +85,7 @@ private:
     void close_stream();    // Closes the stream and resets resources
     void process_stream_cpu(); // Renamed processing function
     void process_stream_gpu(); // Renamed processing function
+    void process_stream_folder();
 
     std::string stream_url_;
     int skip_frame_ = 1;

+ 142 - 0
workspace/folder.json

@@ -0,0 +1,142 @@
+{
+    "models": {
+        "yolo_model_main": {
+            "model_path": "model/model1.engine",
+            "model_type": "YOLO11SEG",
+            "names": [
+                "person",
+                "bicycle",
+                "car",
+                "motorcycle",
+                "airplane",
+                "bus",
+                "train",
+                "truck",
+                "boat",
+                "traffic light",
+                "fire hydrant",
+                "stop sign",
+                "parking meter",
+                "bench",
+                "bird",
+                "cat",
+                "dog",
+                "horse",
+                "sheep",
+                "cow",
+                "elephant",
+                "bear",
+                "zebra",
+                "giraffe",
+                "backpack",
+                "umbrella",
+                "handbag",
+                "tie",
+                "suitcase",
+                "frisbee",
+                "skis",
+                "snowboard",
+                "sports ball",
+                "kite",
+                "baseball bat",
+                "baseball glove",
+                "skateboard",
+                "surfboard",
+                "tennis racket",
+                "bottle",
+                "wine glass",
+                "cup",
+                "fork",
+                "knife",
+                "spoon",
+                "bowl",
+                "banana",
+                "apple",
+                "sandwich",
+                "orange",
+                "broccoli",
+                "carrot",
+                "hot dog",
+                "pizza",
+                "donut",
+                "cake",
+                "chair",
+                "couch",
+                "potted plant",
+                "bed",
+                "dining table",
+                "toilet",
+                "tv",
+                "laptop",
+                "mouse",
+                "remote",
+                "keyboard",
+                "cell phone",
+                "microwave",
+                "oven",
+                "toaster",
+                "sink",
+                "refrigerator",
+                "book",
+                "clock",
+                "vase",
+                "scissors",
+                "teddy bear",
+                "hair drier",
+                "toothbrush"
+            ],
+            "gpu_id": 1,
+            "confidence_threshold": 0.25,
+            "nms_threshold": 0.45
+        }
+    },
+    "pipelines": [
+        {
+            "pipeline_id": "pipeline_1",
+            "description": "处理文件夹图片",
+            "nodes": [
+                {
+                    "node_id": "src_1",
+                    "node_type": "Source",
+                    "params": {
+                        "stream_url": "images/",
+                        "gpu_id": 1,
+                        "decode_type": "GPU",
+                        "skip_frame": 1
+                    }
+                },
+                {
+                    "node_id": "infer_1",
+                    "node_type": "Inference",
+                    "params": {
+                        "model_id": "yolo_model_main"
+                    }
+                },
+                {
+                    "node_id": "track_1",
+                    "node_type": "Tracker",
+                    "params": {
+                        "track_name": "person",
+                        "track_frame": 30,
+                        "track_distance": 30
+                    }
+                },
+                {
+                    "node_id": "analyze_1",
+                    "node_type": "Analyzer",
+                    "params": {}
+                },
+                {
+                    "node_id": "draw_1",
+                    "node_type": "Drawer",
+                    "params": {}
+                },
+                {
+                    "node_id": "push",
+                    "node_type": "Push",
+                    "params": {}
+                }
+            ]
+        }
+    ]
+}