fix

edc5e18b · yhcao6 · 83644927 · edc5e18b
Commit edc5e18b authored Dec 11, 2018 by yhcao6
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

mmdet/models/detectors/two_stage.py mmdet/models/detectors/two_stage.py +2 -2

No files found.
--- a/mmdet/models/detectors/two_stage.py
+++ b/mmdet/models/detectors/two_stage.py
@@ -105,8 +105,8 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
            bbox_assigner = build_assigner(self.train_cfg.rcnn.assigner)
            bbox_sampler = build_sampler(
                self.train_cfg.rcnn.sampler,
-                dict(bbox_roi_extractor=self.bbox_roi_extractor,
+                bbox_roi_extractor=self.bbox_roi_extractor,
-                     bbox_head=self.bbox_head))
+                bbox_head=self.bbox_head)
            num_imgs = img.size(0)
            assign_results = []
            sampling_results = []