-
Notifications
You must be signed in to change notification settings - Fork 52
/
Copy pathmodule.yml
100 lines (93 loc) · 3.4 KB
/
module.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
name: intersection_traffic_meter
parameters:
frame:
width: 1920
height: 1080
output_frame:
codec: ${oc.env:CODEC, 'h264'}
draw_func:
module: samples.intersection_traffic_meter.overlay
class_name: Overlay
kwargs:
config_path: ${oc.env:PROJECT_PATH}/samples/intersection_traffic_meter/polygon_config.yml
detected_object_label: transport
send_stats: True
batch_size: 1
pipeline:
elements:
- element: pyfunc
module: samples.intersection_traffic_meter.line_crossing
class_name: ConditionalDetectorSkip
kwargs:
config_path: ${oc.env:PROJECT_PATH}/samples/intersection_traffic_meter/polygon_config.yml
- element: nvinfer@detector
name: yolov8m
model:
remote:
url: s3://savant-data/models/yolov8m_f630b10/yolov8m_f630b10.zip
checksum_url: s3://savant-data/models/yolov8m_f630b10/yolov8m_f630b10.md5
parameters:
endpoint: https://eu-central-1.linodeobjects.com
format: onnx
model_file: yolov8m.onnx
config_file: config_infer_primary_yoloV8.txt
# max GPU RAM used to build the engine, 6GB by default
# set lower than total GPU RAM available on your hardware
workspace_size: 6144
output:
objects:
# COCO bicycle
- class_id: 1
label: ${parameters.detected_object_label}
selector:
module: savant.selector.detector
class_name: BBoxSelector
kwargs:
confidence_threshold: 0.2
# COCO car
- class_id: 2
label: ${parameters.detected_object_label}
selector:
module: savant.selector.detector
class_name: BBoxSelector
kwargs:
confidence_threshold: 0.2
# COCO motorcycle
- class_id: 3
label: ${parameters.detected_object_label}
selector:
module: savant.selector.detector
class_name: BBoxSelector
kwargs:
confidence_threshold: 0.2
# COCO bus
- class_id: 5
label: ${parameters.detected_object_label}
selector:
module: savant.selector.detector
class_name: BBoxSelector
kwargs:
confidence_threshold: 0.2
# COCO truck
- class_id: 7
label: ${parameters.detected_object_label}
selector:
module: savant.selector.detector
class_name: BBoxSelector
kwargs:
confidence_threshold: 0.2
- element: nvtracker
properties:
ll-lib-file: /opt/nvidia/deepstream/deepstream/lib/libnvds_nvmultiobjecttracker.so
ll-config-file: ${oc.env:PROJECT_PATH}/samples/assets/tracker/config_tracker_NvDCF_perf.yml
tracker-width: 960 # 640 # must be a multiple of 32
tracker-height: 544 # 384
display-tracking-id: 0
- element: pyfunc
module: samples.intersection_traffic_meter.line_crossing
class_name: LineCrossing
kwargs:
config_path: ${oc.env:PROJECT_PATH}/samples/intersection_traffic_meter/polygon_config.yml
stale_track_del_period: 5000
target_obj_label: ${parameters.detected_object_label}
send_stats: ${parameters.send_stats}