-
Notifications
You must be signed in to change notification settings - Fork 1
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
Add export output names specification for individual nodes #112
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. I guess the code will be much nicer after the old way will be deprecated.
Also let's make sure the tests pass merging
Co-authored-by: KlemenSkrlj <[email protected]>
Codecov ReportAttention: Patch coverage is
✅ All tests successful. No failed tests found.
Additional details and impacted files@@ Coverage Diff @@
## main #112 +/- ##
=======================================
Coverage ? 97.07%
=======================================
Files ? 139
Lines ? 6020
Branches ? 0
=======================================
Hits ? 5844
Misses ? 176
Partials ? 0 ☔ View full report in Codecov by Sentry. |
@klemen1999 @kozlov721 archive generation for models with multiple heads should work well now. Please take a look. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
This pull request includes changes to deprecate the
output_names
configuration in favor ofexport_output_names
for more granular control. It updates the documentation, configuration files, and relevant code to reflect this change.Deprecation of
output_names
:configs/README.md
: Markedoutput_names
as deprecated in the documentation.luxonis_train/models/luxonis_lightning.py
: Added a warning message about the deprecation ofexporter.output_names
and implemented logic to usenode.export_output_names
if provided. [1] [2]Configuration updates:
configs/example_export.yaml
: Removedoutput_names
and addedexport_output_names
underhead_params
. [1] [2]tests/configs/archive_config.yaml
: Updated the configuration to useexport_output_names
for various nodes.Code changes:
luxonis_train/nodes/base_node.py
: Addedexport_output_names
as an attribute and updated the constructor and property methods accordingly. [1] [2] [3] [4]luxonis_train/nodes/heads/efficient_bbox_head.py
: Added logic to handleexport_output_names
in the constructor.