-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.yaml
83 lines (72 loc) · 3.08 KB
/
config.yaml
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
######################################
# Ambianic main configuration file #
######################################
version: '2021.02.09'
# path to the data directory
data_dir: ./data
# Set logging level to one of DEBUG, INFO, WARNING, ERROR
logging:
file: ./data/ambianic-log.txt
level: INFO
# set a less noisy log level for the console output
# console_level: WARNING
# Pipeline event timeline configuration
timeline:
event_log: ./data/timeline-event-log.yaml
# Cameras and other input data sources
# Using Home Assistant conventions to ease upcoming integration
sources:
# direct support for raspberry picamera
picamera:
uri: picamera
type: video
live: true
# local video device integration example
webcam:
uri: /dev/video0
type: video
live: true
recorded_cam_feed:
uri: file:///workspace/tests/pipeline/avsource/test2-cam-person1.mkv
# type: video
# live: true
ai_models:
image_detection:
model:
tflite: /opt/ambianic-edge/ai_models/mobilenet_ssd_v2_coco_quant_postprocess.tflite
edgetpu: /opt/ambianic-edge/ai_models/mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite
labels: /opt/ambianic-edge/ai_models/coco_labels.txt
face_detection:
model:
tflite: /opt/ambianic-edge/ai_models/mobilenet_ssd_v2_face_quant_postprocess.tflite
edgetpu: /opt/ambianic-edge/ai_models/mobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite
labels: /opt/ambianic-edge/ai_models/coco_labels.txt
top_k: 2
fall_detection:
model:
tflite: /opt/ambianic-edge/ai_models/posenet_mobilenet_v1_100_257x257_multi_kpt_stripped.tflite
edgetpu: /opt/ambianic-edge/ai_models/posenet_mobilenet_v1_075_721_1281_quant_decoder_edgetpu.tflite
labels: /opt/ambianic-edge/ai_models/pose_labels.txt
# A named pipeline defines an ordered sequence of operations
# such as reading from a data source, AI model inference, saving samples and others.
pipelines:
# Pipeline names could be descriptive, e.g. front_door_watch or entry_room_watch.
area_watch:
- source: picamera
- detect_objects: # run ai inference on the input data
ai_model: image_detection
confidence_threshold: 0.8
# Watch for any of the labels listed below. The labels must be from the model trained label set.
# If no labels are listed, then watch for all model trained labels.
label_filter:
- person
- car
- save_detections: # save samples from the inference results
positive_interval: 300 # how often (in seconds) to save samples with ANY results above the confidence threshold
idle_interval: 6000 # how often (in seconds) to save samples with NO results above the confidence threshold
- detect_falls: # look for falls
ai_model: fall_detection
confidence_threshold: 0.25
- save_detections: # save samples from the inference results
positive_interval: 60 # when a fall is detected, pause for specified number of seconds before looking for another fall
idle_interval: 21600 # save a background image every 6 hours (21600 = 60 seconds * 60 minutes * 6)