fix

xvjiarui · Dec 11, 2018 · edc5e18 · edc5e18
1 parent 8364492
commit edc5e18
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdet/models/detectors/two_stage.py b/mmdet/models/detectors/two_stage.py
@@ -105,8 +105,8 @@ def forward_train(self,
             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_head=self.bbox_head))
+                bbox_roi_extractor=self.bbox_roi_extractor,
+                bbox_head=self.bbox_head)
             num_imgs = img.size(0)
             assign_results = []
             sampling_results = []