Merge pull request #123 from hellock/single-stage
Add SingleStageDetector and RetinaNet
No related branches found
No related tags found
Showing
- .travis.yml 0 additions, 1 deletion.travis.yml
- MODEL_ZOO.md 4 additions, 4 deletionsMODEL_ZOO.md
- README.md 3 additions, 0 deletionsREADME.md
- configs/retinanet_r50_fpn_1x.py 116 additions, 0 deletionsconfigs/retinanet_r50_fpn_1x.py
- mmdet/core/anchor/anchor_target.py 74 additions, 21 deletionsmmdet/core/anchor/anchor_target.py
- mmdet/models/builder.py 7 additions, 2 deletionsmmdet/models/builder.py
- mmdet/models/detectors/__init__.py 4 additions, 2 deletionsmmdet/models/detectors/__init__.py
- mmdet/models/detectors/retinanet.py 14 additions, 0 deletionsmmdet/models/detectors/retinanet.py
- mmdet/models/detectors/single_stage.py 62 additions, 0 deletionsmmdet/models/detectors/single_stage.py
- mmdet/models/rpn_heads/rpn_head.py 2 additions, 2 deletionsmmdet/models/rpn_heads/rpn_head.py
- mmdet/models/single_stage_heads/__init__.py 3 additions, 0 deletionsmmdet/models/single_stage_heads/__init__.py
- mmdet/models/single_stage_heads/retina_head.py 287 additions, 0 deletionsmmdet/models/single_stage_heads/retina_head.py
- mmdet/models/utils/__init__.py 3 additions, 2 deletionsmmdet/models/utils/__init__.py
- mmdet/models/utils/weight_init.py 7 additions, 0 deletionsmmdet/models/utils/weight_init.py
- setup.py 1 addition, 1 deletionsetup.py
Loading
Please register or sign in to comment