Skip to content

GitLab

Unverified Commit 86e25e41 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #74 from hellock/hotfix 

bug fix for missing weight initialization
parents 573d4c33 69eedb8a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
mmdet/models/detectors/two_stage.py
View file @ 86e25e41
...@@ -65,6 +65,9 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin, ...@@ -65,6 +65,9 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
if self.with_bbox: if self.with_bbox:
self.bbox_roi_extractor.init_weights() self.bbox_roi_extractor.init_weights()
self.bbox_head.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): def extract_feat(self, img):
x = self.backbone(img) x = self.backbone(img)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment