bug fix for missing weight initialization

69eedb8a · Kai Chen · d6abbef4 · 69eedb8a
Commit 69eedb8a authored Oct 25, 2018 by Kai Chen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

mmdet/models/detectors/two_stage.py mmdet/models/detectors/two_stage.py +3 -0

No files found.
--- a/mmdet/models/detectors/two_stage.py
+++ b/mmdet/models/detectors/two_stage.py
@@ -65,6 +65,9 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
        if self.with_bbox:
            self.bbox_roi_extractor.init_weights()
            self.bbox_head.init_weights()
+        if self.with_mask:
+            self.mask_roi_extractor.init_weights()
+            self.mask_head.init_weights()

    def extract_feat(self, img):
        x = self.backbone(img)