search

whizzmobility / models

Models and examples built with TensorFlow. Added support for more recent SOTA, and implementing on private datasets.
Apache License 2.0
0 stars 0 forks source link

[Calibrate] Calibrate NMS #100

Open rehohoho opened 3 years ago

rehohoho commented 3 years ago

Calibrate thresholds for bounding boxes:

Code reference: https://github.com/whizzmobility/models/blob/master/official/vision/beta/serving/multitask.py#L132

TF NMS implementation docs: https://www.tensorflow.org/api_docs/python/tf/image/non_max_suppression

rehohoho commented 3 years ago

Awaiting new model with improvement from #99.