Skip to content
Permalink
master
Switch branches/tags
Go to file
 
 
Cannot retrieve contributors at this time

Config Format: YAML

Config Example

Path to Model's .pb file Supported Models
weights: "frozen_inference_graph.pb"
Threshold [0.0, 1.0]
threshold: 0.5
Input Shape
input_shape: [w, h]
Log device placement
log_device_placement: true or false
GPU Options:
  • explained

    per_process_gpu_memory_fraction: 0.0
    
device: "GPU|CPU", "GPU", "CPU", "GPU:0", ..

Labels Format

labels: mscoco_label_map.yml or {1: "person", ..}