add model dimensions to config

This commit is contained in:
Blake Blackshear
2020-12-08 19:39:46 -06:00
parent 9c79392060
commit 5053305e17
2 changed files with 38 additions and 0 deletions

View File

@@ -199,6 +199,13 @@ detectors:
# Optional: device name as defined here: https://coral.ai/docs/edgetpu/multiple-edgetpu/#using-the-tensorflow-lite-python-api
device: usb
# Optional: model configuration
model:
# Required: height of the trained model
height: 320
# Required: width of the trained model
width: 320
# Required: mqtt configuration
mqtt:
# Required: host name
@@ -880,6 +887,8 @@ Models for both CPU and EdgeTPU (Coral) are bundled in the image. You can use yo
- EdgeTPU Model: `/edgetpu_model.tflite`
- Labels: `/labelmap.txt`
You also need to update the model width/height in the config if they differ from the defaults.
### Customizing the Labelmap
The labelmap can be customized to your needs. A common reason to do this is to combine multiple object types that are easily confused when you don't need to be as granular such as car/truck. You must retain the same number of labels, but you can change the names. To change: