Skip to content

Support yololite with fused NMS#2203

Open
leeclemnet wants to merge 4 commits intomainfrom
feat/yololite-fused-nms
Open

Support yololite with fused NMS#2203
leeclemnet wants to merge 4 commits intomainfrom
feat/yololite-fused-nms

Conversation

@leeclemnet
Copy link
Copy Markdown
Contributor

@leeclemnet leeclemnet commented Apr 6, 2026

What does this PR do?

roboflow/yololite#7 and https://github.com/roboflow/roboflow-train/pull/763 extend the yololite trainer to export the same 3 ONNX variants as other yolos: (1) static batch size non-fused NMS, (2) dynamic batch size non-fused NMS, (3) 'dynamic' batch size fused NMS.

yololite already exports and registers (2), which inference_models has been using so far, this PR adds support for (1) and (3). The change is backwards compatible so already-trained and exported yololite still work (see testing).

This PR can be merged independently of https://github.com/roboflow/roboflow-train/pull/763.

Type of Change

  • New feature (non-breaking change that adds functionality)

Testing

  • I have tested this change locally
  • I have added/updated tests for this change

Test details:

  • Coin counting integration tests for the 3 different onnx variants that yololite will export after https://github.com/roboflow/roboflow-train/pull/763
  • Tested previously exported yololite onnx model against local inference server in workflows - backwards compatible
image
  • Tested new model version against local inference server in workflows - similar output
image
  • Tested new model version against current serverless inference in workflows - still works, same output as local
image

Checklist

  • My code follows the style guidelines of this project
  • I have performed a self-review of my own code
  • I have commented my code where necessary, particularly in hard-to-understand areas
  • My changes generate no new warnings or errors
  • I have updated the documentation accordingly (if applicable)

Additional Context

@leeclemnet leeclemnet force-pushed the feat/yololite-fused-nms branch from 2071af7 to 7250082 Compare April 6, 2026 17:46
@leeclemnet leeclemnet changed the title Support yololite with fused NMS Optionally export ONNX with fused NMS Apr 6, 2026
@leeclemnet leeclemnet changed the title Optionally export ONNX with fused NMS Support yololite with fused NMS Apr 6, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant