Description

This PR updates all the ops needed for the YOLO detection generator and adds an nms_ops file that holds the nms used for box filtering on the TPU. This PR adds DIOU nms as well as a quick NMS that is used for compiling the model in tflite.

Type of change

For a new feature or function, please create an issue first to discuss it
with us before submitting a pull request.

Note: Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• Documentation update
• TensorFlow 2 migration
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• A new research paper code implementation
• Other (Specify)

Tests

box op and math op test cases

python3.8 -m official.vision.beta.projects.yolo.ops.box_ops_test

nms op tests will be included in the next commit

Test Configuration:

Checklist

• I have signed the Contributor License Agreement.
• I have read guidelines for pull request.
• My code follows the coding guidelines.
• I have performed a self code review of my own code.
• I have commented my code, particularly in hard-to-understand areas.
• I have made corresponding changes to the documentation.
• My changes generate no new warnings.
• I have added tests that prove my fix is effective or that my feature works.