-
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 gradient visualization during validation #115
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.
Overall LGTM, left some comments
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
Codecov ReportAttention: Patch coverage is
✅ All tests successful. No failed tests found.
Additional details and impacted files@@ Coverage Diff @@
## main #115 +/- ##
==========================================
- Coverage 97.12% 96.31% -0.81%
==========================================
Files 139 147 +8
Lines 6012 6304 +292
==========================================
+ Hits 5839 6072 +233
- Misses 173 232 +59 ☔ View full report in Codecov by Sentry. |
This pull request introduces a new callback for visualizing gradients using Grad-CAM during validation in the
luxonis_train
package. The most important changes include the addition of theGradCamCallback
class, updates to the callback registration, and documentation updates.New Callback Implementation:
luxonis_train/callbacks/gradcam_visializer.py
: Added theGradCamCallback
class and its supportingModelWrapper
class to enable gradient visualization using Grad-CAM during validation.Callback Registration:
luxonis_train/callbacks/__init__.py
: Registered theGradCamCallback
class to make it available for use in the training pipeline.Documentation:
luxonis_train/callbacks/README.md
: Updated the documentation to include a description of theGradCamCallback
and its parameters.