bigmlcom / sensenet

0 stars 8 forks source link

Allow multiple boxes in tflite yolo models #28

Closed charleslparker closed 2 years ago

charleslparker commented 2 years ago

TFLite really dislikes tensors with unknown dimensions, so we have to utilize the padding option in non-max suppression to get tensors with fixed dimensions everywhere. Otherwise, TFLite models will only give the maximum score box.