-
Notifications
You must be signed in to change notification settings - Fork 9.5k
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
[Enhancement]: Refactor SSD #5291
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5291 +/- ##
==========================================
- Coverage 65.14% 64.98% -0.16%
==========================================
Files 275 276 +1
Lines 21248 21289 +41
Branches 3535 3540 +5
==========================================
- Hits 13841 13835 -6
- Misses 6654 6698 +44
- Partials 753 756 +3
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
Can we add a model conversion script that can load the previously trained SSD successfully loaded in this PR? We can also do that through |
# Conflicts: # configs/ssd/README.md # tests/test_models/test_necks.py
Since the extra layers in SSDVGG have been moved to SSDNeck, the _load_from_state_dict function needs to be put into the detector class. But SSD uses SingleStageDetector so I choose to write a convert weight script instead of putting the weight convert function into the base class. |
We also need to update the compatibility.md |
This PR including these modifications
I have trained SSD300 and the mAP is 25.9 (origin 25.6), SSD512 mAP is 29.8 (origin 29.4).