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

Enhance Filtering of Annotations by Model and Class in BaseFormat #7

Merged
merged 13 commits into from
Sep 10, 2024

Conversation

shamspias
Copy link
Owner

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:

  • Model-Specific Annotation Processing: Updated the 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).
  • Conditional Class Inclusion: Annotations for each class are now included only if the class identifier exists within the supported_classes list derived from YAML configuration files. This prevents irrelevant class annotations from being processed.

Benefits:

  • Enhanced Accuracy and Relevance: By ensuring annotations are only created for supported classes and specific model types, the quality and applicability of training data are significantly improved.
  • Increased Flexibility and Configurability: Facilitates easy adaptation to different models and class sets by adjusting YAML configurations, enhancing the system's usability across various datasets and requirements.
  • Streamlined Annotation Process: Reduces computational overhead and potential errors by filtering out irrelevant classes early in the annotation process.

@shamspias shamspias merged commit 496df74 into main Sep 10, 2024
1 check passed
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