Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
32 lines (17 sloc) 921 Bytes

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: "/device:GPU:0"  # "/device:CPU:0"

Labels Format

labels: mscoco_label_map.yml
You can’t perform that action at this time.