-
Notifications
You must be signed in to change notification settings - Fork 0
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
Added dai_type to NNArchive #200
base: main
Are you sure you want to change the base?
Conversation
Codecov ReportAll modified and coverable lines are covered by tests ✅
✅ All tests successful. No failed tests found. Additional details and impacted files@@ Coverage Diff @@
## main #200 +/- ##
==========================================
+ Coverage 92.33% 92.52% +0.19%
==========================================
Files 79 79
Lines 4161 4227 +66
==========================================
+ Hits 3842 3911 +69
+ Misses 319 316 -3 ☔ View full report in Codecov by Sentry. |
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.
Left a small comment, otherwise LGTM
deprecated="Deprecated, use `dai_type` instead.", | ||
description="If True input to the model is interleaved (NHWC) else planar (NCHW).", | ||
) | ||
dai_type: str = Field( |
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.
We should probably validate the value
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.
+1 on validating this str value. Maybe have a list of all the possible data types from DAI and ensure the given value exists there?
luxonis_ml/nn_archive/config_building_blocks/base_models/input.py
Outdated
Show resolved
Hide resolved
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.
Just left a couple of comments, otherwise LGTM.
Main changes:
config_version
is a string in format "x.y" where x and y are integerssubtype
is a string and not an Enumreverse_input_channels
andinterleaved_to_planar
to match their real meaningdai_type
to the archive which is what DAI will read to automatically setup the pipeline