There was an error fetching the commit references. Please try again later.
Allow for images to contain zero true detections (#1531)
* Allow for images to contain zero true detections * Allow for empty assignment in PointAssigner * Allow ApproxMaxIouAssigner to return an empty result * Fix CascadeRNN forward when entire batch has no truth * Correctly assign boxes to background when there is no truth * Fix assignment tests * Make flatten robust * Fix bbox loss with empty pred/truth * Fix logic error in BBoxHead.loss * Add tests for empty truth cases * tests faster rcnn empty forward * Skip roipool forward tests if torchvision is not installed * Add tests for bbox/anchor heads * Consolidate test_forward and test_forward2 * Fix assign_results.labels = None when gt_labels is given; Add test for this case * Fix OHEM Sampler with zero truth * remove xdev * resolve 3 reviews * Fix flake8 * refactoring * fix yaml format * add filter flag * minor fix * delete redundant code in load anno * fix flake8 errors * quick fix for empty truth with masks * fix yapf error * fix mask padding for empty masks Co-authored-by:Cao Yuhang <yhcao6@gmail.com> Co-authored-by:
Kai Chen <chenkaidev@gmail.com>
Showing
- mmdet/core/bbox/assigners/approx_max_iou_assigner.py 9 additions, 5 deletionsmmdet/core/bbox/assigners/approx_max_iou_assigner.py
- mmdet/core/bbox/assigners/assign_result.py 74 additions, 1 deletionmmdet/core/bbox/assigners/assign_result.py
- mmdet/core/bbox/assigners/max_iou_assigner.py 26 additions, 5 deletionsmmdet/core/bbox/assigners/max_iou_assigner.py
- mmdet/core/bbox/assigners/point_assigner.py 17 additions, 3 deletionsmmdet/core/bbox/assigners/point_assigner.py
- mmdet/core/bbox/geometry.py 28 additions, 3 deletionsmmdet/core/bbox/geometry.py
- mmdet/core/bbox/samplers/base_sampler.py 1 addition, 1 deletionmmdet/core/bbox/samplers/base_sampler.py
- mmdet/datasets/coco.py 1 addition, 1 deletionmmdet/datasets/coco.py
- mmdet/datasets/custom.py 4 additions, 2 deletionsmmdet/datasets/custom.py
- mmdet/datasets/pipelines/loading.py 1 addition, 14 deletionsmmdet/datasets/pipelines/loading.py
- mmdet/datasets/pipelines/transforms.py 4 additions, 1 deletionmmdet/datasets/pipelines/transforms.py
- mmdet/models/bbox_heads/bbox_head.py 22 additions, 18 deletionsmmdet/models/bbox_heads/bbox_head.py
- mmdet/models/bbox_heads/convfc_bbox_head.py 5 additions, 3 deletionsmmdet/models/bbox_heads/convfc_bbox_head.py
- mmdet/models/detectors/cascade_rcnn.py 6 additions, 0 deletionsmmdet/models/detectors/cascade_rcnn.py
- mmdet/models/detectors/two_stage.py 9 additions, 9 deletionsmmdet/models/detectors/two_stage.py
- tests/test_assigner.py 261 additions, 0 deletionstests/test_assigner.py
- tests/test_forward.py 158 additions, 0 deletionstests/test_forward.py
- tests/test_heads.py 171 additions, 0 deletionstests/test_heads.py
- tests/test_sampler.py 235 additions, 0 deletionstests/test_sampler.py
Loading
Please register or sign in to comment