Enable TensorFlow ops for --nms
and --agnostic-nms
#7281
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
π οΈ PR Summary
Made with β€οΈ by Ultralytics Actions
π Summary
Enhancements to TensorFlow Lite (TFLite) export functionality in YOLOv5.
π Key Changes
nms
andagnostic_nms
arguments toexport_tflite
function.ncalib
to a default value of 100.tf.lite.OpsSet.SELECT_TF_OPS
to supported operations whennms
oragnostic_nms
is enabled during TFLite export.π― Purpose & Impact
nms
(Non-Maximum Suppression) andagnostic_nms
parameters aims to allow users more control over the post-processing steps during TFLite model conversion, potentially leading to more efficient on-device inference.ncalib
value helps standardize the quantization calibration step, which can improve the quality of the int8 quantized models.SELECT_TF_OPS
supports a wider range of TensorFlow operations when exporting to TFLite, thereby enhancing model compatibility and performance on a broader set of devices.