Enhance Filtering of Annotations by Model and Class in BaseFormat #7
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.
Summary:
This merge request introduces enhancements to the
BaseFormat
class, adding the capability to filter annotations based on model type (RTDETR, NAS, YOLO) and ensure that only classes specified in the project's YAML configurations are annotated. This change ensures annotations are not only accurate but also relevant to the specific models and use cases.Changes:
process_results
method to incorporate checks that validate the model type before processing annotations. This ensures that annotations are generated appropriately according to the model type used (RTDETR, NAS, YOLO).supported_classes
list derived from YAML configuration files. This prevents irrelevant class annotations from being processed.Benefits: