@@ -1,9 +1,10 @@
 from ..builder import DETECTORS
 from .single_stage import SingleStageDetector
+from .single_stage_test_mixins import BBoxTestMixin
 
 
 @DETECTORS.register_module()
-class RetinaNet(SingleStageDetector):
+class RetinaNet(SingleStageDetector, BBoxTestMixin):
     """Implementation of `RetinaNet <https://arxiv.org/abs/1708.02002>`_"""
 
     def __init__(self,
from ..builder import DETECTORS
from .single_stage import SingleStageDetector


@DETECTORS.register_module()
class RetinaNet(SingleStageDetector):
    """Implementation of `RetinaNet <https://arxiv.org/abs/1708.02002>`_"""

    def __init__(self,
                 backbone,
                 neck,
                 bbox_head,
                 train_cfg=None,
                 test_cfg=None,
                 pretrained=None):
        super(RetinaNet, self).__init__(backbone, neck, bbox_head, train_cfg,
                                        test_cfg, pretrained)

To be consistent with two-stage detectors, it is better to make BBoxTestMixin a mixin of DenseHead instead of Detector.