0.8 beta 2: custom model crashes: IndexError: index 20 is out of bounds for axis 0 with size 20,

Describe the bug Using a model I previously trained (and that worked in doods): File “sqwrlmodel2.tflite” from https://github.com/rb2k/sqwrlmodel

Dimensions are 320x320 according to https://github.com/rb2k/sqwrlmodel/blob/main/sqwrlmodel2_metadata.json

Dictionary is:

0 person
1 cat
2 bird
3 squirrel
4 dog

Version of frigate 0.8b2

Config file Include your full config file wrapped in triple back ticks.

model:
  # Required: height of the trained model
  height: 320
  # Required: width of the trained model
  width: 320

Logs

Process detector:coral:,
Traceback (most recent call last):,
  File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap,
    self.run(),
  File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run,
    self._target(*self._args, **self._kwargs),
  File "/opt/frigate/frigate/edgetpu.py", line 149, in run_detector,
    detections = object_detector.detect_raw(input_frame),
  File "/opt/frigate/frigate/edgetpu.py", line 105, in detect_raw,
    detections[i] = [label_codes[i], score, boxes[i][0], boxes[i][1], boxes[i][2], boxes[i][3]],
IndexError: index 20 is out of bounds for axis 0 with size 20,

Computer Hardware

• OS: OpenSuse
• Virtualization: Docker
• Coral Version: USB
• Network Setup: Wired

[blakeblackshear]

Looks like your output shape is different.

[stale[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.