Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Convert input for quantized YOLOv8 #1521

Merged
merged 8 commits into from
May 4, 2023

Conversation

anmarques
Copy link
Member

Added call to function that skips the quantization of the input if the model is quantized. After that it expects the input to be quantized.

Testing plan:
Exported YOLOv8s dense quantized. Verified that the input is quantized and the QuantizationLinear node is removed. Tested for accuracy and latency on deepsparse.

@anmarques anmarques requested review from bfineran and KSGulin April 13, 2023 21:12
bfineran
bfineran previously approved these changes Apr 13, 2023
@anmarques anmarques requested a review from natuan April 14, 2023 14:21
@anmarques anmarques merged commit 7bf70d2 into main May 4, 2023
@anmarques anmarques deleted the fix/yolov8/skip_onnx_input_quantize branch May 4, 2023 18:13
bfineran pushed a commit that referenced this pull request Jun 15, 2023
* Added call to function that skips the quantization of the input if the model is quantized

* Removed bare exception

* Move input data type conversion upstream so it is properly saved and used as input to ort

* Style and quality fixes

---------

Co-authored-by: Michael Goin <[email protected]>
bfineran added a commit that referenced this pull request Jun 15, 2023
* Added call to function that skips the quantization of the input if the model is quantized

* Removed bare exception

* Move input data type conversion upstream so it is properly saved and used as input to ort

* Style and quality fixes

---------

Co-authored-by: Alexandre Marques <[email protected]>
Co-authored-by: Michael Goin <[email protected]>
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.

4 participants