diff --git a/configs/sample_person_shoe_two_stage_balanced_ffmpeg_decode.json b/configs/sample_person_shoe_two_stage_balanced_ffmpeg_decode.json new file mode 100644 index 0000000..5d65841 --- /dev/null +++ b/configs/sample_person_shoe_two_stage_balanced_ffmpeg_decode.json @@ -0,0 +1,170 @@ +{ + "queue": { + "size": 8, + "strategy": "drop_oldest" + }, + "graphs": [ + { + "name": "person_shoe_two_stage_balanced_ffmpeg_decode", + "executor": { + "batch_size": 2, + "run_budget": 8 + }, + "nodes": [ + { + "id": "in", + "type": "input_rtsp", + "role": "source", + "enable": true, + "url": "rtsp://10.0.0.49:8554/cam", + "fps": 30, + "width": 1920, + "height": 1080, + "use_ffmpeg": true, + "use_mpp": false, + "force_tcp": true + }, + { + "id": "pre_rgb", + "type": "preprocess", + "role": "filter", + "enable": true, + "dst_w": 1920, + "dst_h": 1080, + "dst_format": "rgb", + "dst_packed": true, + "resize_mode": "stretch", + "rga_gate": "person_shoe_two_stage_balanced_ffmpeg_decode", + "use_rga": true + }, + { + "id": "person_det", + "type": "ai_yolo", + "role": "filter", + "enable": true, + "cpu_affinity": [4], + "use_rga": true, + "rga_gate": "person_shoe_two_stage_balanced_ffmpeg_decode", + "rga_max_inflight": 4, + "dst_packed": true, + "use_dma_input": true, + "infer_fps": 2, + "infer_phase_ms": 0, + "model_path": "./models/yolov8n-640.rknn", + "model_version": "v8", + "model_w": 640, + "model_h": 640, + "num_classes": 80, + "conf": 0.35, + "nms": 0.45, + "class_filter": [0], + "bbox_expand": { + "enable": true, + "class_id": 0, + "left": 0.06, + "right": 0.06, + "top": 0.04, + "bottom": 0.16 + } + }, + { + "id": "person_trk", + "type": "tracker", + "role": "filter", + "enable": true, + "cpu_affinity": [5], + "mode": "bytetrack_lite", + "per_class": true, + "track_classes": [0], + "high_th": 0.6, + "low_th": 0.15, + "iou_th": 0.3, + "max_age_ms": 700, + "min_hits": 2 + }, + { + "id": "shoe_det", + "type": "ai_shoe_det", + "role": "filter", + "enable": true, + "cpu_affinity": [6], + "use_rga": true, + "rga_gate": "person_shoe_two_stage_balanced_ffmpeg_decode", + "rga_max_inflight": 4, + "dst_packed": true, + "use_dma_input": false, + "infer_fps": 2, + "infer_phase_ms": 250, + "model_path": "./models/shoe_detector_openimages_ppe_v1.rknn", + "model_w": 640, + "model_h": 640, + "conf": 0.20, + "nms": 0.45, + "v8_box_format": "cxcywh", + "append_detections": true, + "dynamic_roi": { + "enable": true, + "person_class_id": 0, + "shoe_class_id": 1, + "debug_roi_class_id": -1, + "max_rois": 1, + "min_person_height": 80, + "x_offset": -0.20, + "y_offset": 0.68, + "width_scale": 1.40, + "height_scale": 0.50 + } + }, + { + "id": "osd", + "type": "osd", + "role": "filter", + "enable": true, + "cpu_affinity": [7], + "draw_bbox": true, + "draw_text": false, + "use_rga_bbox": false, + "labels": ["person", "shoe"] + }, + { + "id": "post", + "type": "preprocess", + "role": "filter", + "enable": true, + "cpu_affinity": [7], + "dst_w": 1920, + "dst_h": 1080, + "dst_format": "nv12", + "resize_mode": "stretch", + "rga_gate": "person_shoe_two_stage_balanced_ffmpeg_decode", + "use_rga": true + }, + { + "id": "pub", + "type": "publish", + "role": "filter", + "enable": true, + "cpu_affinity": [3], + "codec": "h264", + "fps": 30, + "bitrate_kbps": 2000, + "mpp_output_timeout_ms": 50, + "mpp_packet_wait_ms": 10, + "use_mpp": true, + "outputs": [ + {"proto": "rtsp_server", "port": 8555, "path": "/live/cam1"} + ] + } + ], + "edges": [ + ["in", "pre_rgb"], + ["pre_rgb", "person_det"], + ["person_det", "person_trk"], + ["person_trk", "shoe_det", {"queue": {"size": 16, "strategy": "drop_oldest"}}], + ["shoe_det", "osd", {"queue": {"size": 16, "strategy": "drop_oldest"}}], + ["osd", "post", {"queue": {"size": 32, "strategy": "drop_oldest"}}], + ["post", "pub", {"queue": {"size": 64, "strategy": "drop_oldest"}}] + ] + } + ] +} diff --git a/configs/sample_rtsp_publish_only_ffmpeg_decode.json b/configs/sample_rtsp_publish_only_ffmpeg_decode.json new file mode 100644 index 0000000..b2e3905 --- /dev/null +++ b/configs/sample_rtsp_publish_only_ffmpeg_decode.json @@ -0,0 +1,49 @@ +{ + "queue": { + "size": 16, + "strategy": "drop_oldest" + }, + "graphs": [ + { + "name": "rtsp_publish_only_ffmpeg_decode", + "executor": { + "batch_size": 2, + "run_budget": 8 + }, + "nodes": [ + { + "id": "in", + "type": "input_rtsp", + "role": "source", + "enable": true, + "url": "rtsp://10.0.0.49:8554/cam", + "fps": 30, + "width": 1920, + "height": 1080, + "use_ffmpeg": true, + "use_mpp": false, + "force_tcp": true + }, + { + "id": "pub", + "type": "publish", + "role": "filter", + "enable": true, + "cpu_affinity": [3], + "codec": "h264", + "fps": 30, + "bitrate_kbps": 2000, + "mpp_output_timeout_ms": 50, + "mpp_packet_wait_ms": 10, + "use_mpp": true, + "outputs": [ + {"proto": "rtsp_server", "port": 8555, "path": "/live/cam1"} + ] + } + ], + "edges": [ + ["in", "pub", {"queue": {"size": 64, "strategy": "drop_oldest"}}] + ] + } + ] +}