bug fix for missing weight initialization

Author: hellock

mmdet/models/detectors/two_stage.py

@@ -65,6 +65,9 @@ def init_weights(self, pretrained=None):
         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)