-
Notifications
You must be signed in to change notification settings - Fork 150
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 YOLOF mmdet model #1030
Add YOLOF mmdet model #1030
Conversation
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Codecov Report
@@ Coverage Diff @@
## master #1030 +/- ##
==========================================
+ Coverage 85.71% 85.80% +0.09%
==========================================
Files 258 261 +3
Lines 5679 5694 +15
==========================================
+ Hits 4868 4886 +18
+ Misses 811 808 -3
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
Thanks for your contribution @dnth ! |
I am so happy as this is my first contribution to IceVision! Thank you for the guides and documentations! |
@@ -47,7 +47,7 @@ def param_groups(model): | |||
layers += [getattr(body, l) for l in body.res_layers] | |||
|
|||
# add the neck module if it exists (DETR doesn't have a neck module) | |||
layers += [model.neck for name in model.named_modules() if name == "neck"] | |||
layers += [module for name, module in model.named_modules() if name == "neck"] |
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.
This change was merged in this PR
This PR adds the YOLOF model from mmdet to IceVision