| from ..builder import DETECTORS |
| from .single_stage import SingleStageDetector |
|
|
|
|
| @DETECTORS.register_module() |
| class RepPointsDetector(SingleStageDetector): |
| """RepPoints: Point Set Representation for Object Detection. |
| |
| This detector is the implementation of: |
| - RepPoints detector (https://arxiv.org/pdf/1904.11490) |
| """ |
|
|
| def __init__(self, |
| backbone, |
| neck, |
| bbox_head, |
| train_cfg=None, |
| test_cfg=None, |
| pretrained=None): |
| super(RepPointsDetector, |
| self).__init__(backbone, neck, bbox_head, train_cfg, test_cfg, |
| pretrained) |
|
|
