-
Notifications
You must be signed in to change notification settings - Fork 39
/
module.yml
148 lines (139 loc) · 5.39 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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
name: ${oc.env:MODULE_NAME, 'traffic-meter'}
parameters:
frame:
width: 1280
height: 720
padding:
# Paddings are kept on the output frame
keep: true
left: 0
right: 0
# Padding size is 180px on the top of the frame
top: 180
bottom: 0
geometry_base: 4
output_frame:
codec: ${oc.env:CODEC, 'h264'}
draw_func:
module: samples.traffic_meter.overlay
# specify the drawfunc's python class from the module
class_name: Overlay
kwargs:
overlay_height: 180
detected_object:
id: 0
label: person
batch_size: 1
detector: ${oc.env:DETECTOR, 'yolov8m'}
send_stats: True
pipeline:
elements:
# regular pyfunc unit definition
- element: pyfunc
module: samples.traffic_meter.line_crossing
class_name: ConditionalDetectorSkip
kwargs:
config_path: ${oc.env:PROJECT_PATH}/samples/traffic_meter/line_config.yml
# variant group: peoplenet detector
# groups are included in the pipeline.elements alongside regular units or other groups
- group:
# enabled if env var DETECTOR==peoplenet
init_condition:
expr: ${parameters.detector}
value: peoplenet
elements:
# below follows detector unit definition as usual
- element: nvinfer@detector
name: peoplenet
model:
format: etlt
remote:
url: s3://savant-data/models/peoplenet/peoplenet_pruned_v2.0.zip
checksum_url: s3://savant-data/models/peoplenet/peoplenet_pruned_v2.0.md5
parameters:
endpoint: https://eu-central-1.linodeobjects.com
model_file: resnet34_peoplenet_pruned.etlt
batch_size: ${parameters.batch_size}
input:
layer_name: input_1
shape: [3, 544, 960]
scale_factor: 0.0039215697906911373
output:
layer_names: [output_bbox/BiasAdd, output_cov/Sigmoid]
num_detected_classes: 3
objects:
- class_id: ${parameters.detected_object.id}
label: ${parameters.detected_object.label}
# variant group: yolov8m detector
# groups are included in the pipeline.elements alongside regular units or other groups
- group:
# enabled if env var DETECTOR==yolov8m
init_condition:
expr: ${parameters.detector}
value: yolov8m
elements:
# below follows detector unit definition as usual
- 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
batch_size: ${parameters.batch_size}
# 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:
- class_id: ${parameters.detected_object.id}
label: ${parameters.detected_object.label}
# variant group: yolov8s detector
# groups are included in the pipeline.elements alongside regular units or other groups
- group:
# enabled if env var DETECTOR==yolov8s
init_condition:
expr: ${parameters.detector}
value: yolov8s
elements:
# below follows detector unit definition as usual
- element: nvinfer@detector
name: yolov8s
model:
remote:
url: s3://savant-data/models/yolov8s_f630b10/yolov8s_f630b10.zip
checksum_url: s3://savant-data/models/yolov8s_f630b10/yolov8s_f630b10.md5
parameters:
endpoint: https://eu-central-1.linodeobjects.com
format: onnx
model_file: yolov8s.onnx
config_file: config_infer_primary_yoloV8.txt
batch_size: ${parameters.batch_size}
# 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:
- class_id: ${parameters.detected_object.id}
label: ${parameters.detected_object.label}
# regular nvtracker definition
- 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
# regular pyfunc unit definition
- element: pyfunc
module: samples.traffic_meter.line_crossing
class_name: LineCrossing
kwargs:
config_path: ${oc.env:PROJECT_PATH}/samples/traffic_meter/line_config.yml
stale_track_del_period: 5000
target_obj_label: ${parameters.detected_object.label}
send_stats: ${parameters.send_stats}