forked from gustavz/realtime_object_detection
-
Notifications
You must be signed in to change notification settings - Fork 9
/
config.yml
16 lines (15 loc) · 792 Bytes
/
config.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
---
video_input: 0 # Input Must be OpenCV readable
visualize: True # Disable for performance increase
max_frames: 500 # only used if visualize==False
width: 300 # 300x300 is used by SSD_Mobilenet -> highest fps
height: 300
fps_interval: 5 # Intervall [s] to print fps in console
allow_memory_growth: True # restart python to apply changes on memory usage
det_interval: 75 # intervall [frames] to print detections to console
det_th: 0.5 # detection threshold for det_intervall
# Change only if you use another model
model_name: 'ssd_mobilenet_v11_coco'
model_path: 'models/ssd_mobilenet_v11_coco/frozen_inference_graph.pb'
label_path: 'object_detection/data/mscoco_label_map.pbtxt'
num_classes: 90