Skip to content

Support yololite object detection in inference_models with ONNX backend#2078

Merged
leeclemnet merged 6 commits intomainfrom
feat/yololite-inference-models
Mar 5, 2026
Merged

Support yololite object detection in inference_models with ONNX backend#2078
leeclemnet merged 6 commits intomainfrom
feat/yololite-inference-models

Conversation

@leeclemnet
Copy link
Contributor

@leeclemnet leeclemnet commented Mar 4, 2026

What does this PR do?

Adds support for yololite object detection in inference_models, ONNX backend only to start

We will not do the same for legacy inference, since traffic is no longer being routed there.

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:

  • Added integration test with canned model package
  • Tested inferencing yololite-edge-n model trained on platform (staging) using local build of x86.cu124.base.dockerfile - outputs as expected
image
  • Tested Object Detection workflow block with yololite-edge-n model with local Dockerfile.onnx.gpu.dev build and USE_INFERENCE_MODELS=True
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 changed the title Support yololite object detection in inference_models with ONNX Support yololite object detection in inference_models with ONNX backend Mar 4, 2026
@leeclemnet leeclemnet force-pushed the feat/yololite-inference-models branch from 502911d to 6f4bbf3 Compare March 4, 2026 20:21
Copy link
Collaborator

@PawelPeczek-Roboflow PawelPeczek-Roboflow left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM so far. -let me know do you need anything as support moving forward

@leeclemnet leeclemnet force-pushed the feat/yololite-inference-models branch from 6f4bbf3 to 30ea158 Compare March 5, 2026 13:43
@leeclemnet leeclemnet marked this pull request as ready for review March 5, 2026 14:34
@PawelPeczek-Roboflow PawelPeczek-Roboflow self-requested a review March 5, 2026 15:01
@leeclemnet leeclemnet merged commit 3282190 into main Mar 5, 2026
56 checks passed
@leeclemnet leeclemnet deleted the feat/yololite-inference-models branch March 5, 2026 15:03
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.

2 participants