Add flops counter (#1127)
* add flops counter * minor fix * add forward_dummy() for most detectors * add documentation for some tools
Showing
- GETTING_STARTED.md 97 additions, 0 deletionsGETTING_STARTED.md
- demo/loss_curve.png 0 additions, 0 deletionsdemo/loss_curve.png
- mmdet/models/detectors/cascade_rcnn.py 31 additions, 0 deletionsmmdet/models/detectors/cascade_rcnn.py
- mmdet/models/detectors/double_head_rcnn.py 24 additions, 0 deletionsmmdet/models/detectors/double_head_rcnn.py
- mmdet/models/detectors/grid_rcnn.py 25 additions, 0 deletionsmmdet/models/detectors/grid_rcnn.py
- mmdet/models/detectors/htc.py 40 additions, 0 deletionsmmdet/models/detectors/htc.py
- mmdet/models/detectors/mask_scoring_rcnn.py 3 additions, 0 deletionsmmdet/models/detectors/mask_scoring_rcnn.py
- mmdet/models/detectors/rpn.py 5 additions, 0 deletionsmmdet/models/detectors/rpn.py
- mmdet/models/detectors/single_stage.py 5 additions, 0 deletionsmmdet/models/detectors/single_stage.py
- mmdet/models/detectors/two_stage.py 29 additions, 0 deletionsmmdet/models/detectors/two_stage.py
- mmdet/utils/__init__.py 2 additions, 1 deletionmmdet/utils/__init__.py
- mmdet/utils/flops_counter.py 421 additions, 0 deletionsmmdet/utils/flops_counter.py
- tools/get_flops.py 52 additions, 0 deletionstools/get_flops.py
Loading
Please register or sign in to comment