Update maskformer.py and Update maskformer_head.py (#8263)

* Update maskformer.py

* Update maskformer_head.py

* Update maskformer.py

* fix lint

* fix lint

Co-authored-by: Cedric Luo <[email protected]>

Author: duanzhihua

mmdet/models/dense_heads/maskformer_head.py

@@ -116,10 +116,11 @@ def __init__(self,
         self.test_cfg = test_cfg
         self.train_cfg = train_cfg
         if train_cfg:
-            self.assigner = build_assigner(train_cfg.assigner)
-            self.sampler = build_sampler(train_cfg.sampler, context=self)
+            self.assigner = build_assigner(train_cfg.get('assigner', None))
+            self.sampler = build_sampler(
+                train_cfg.get('sampler', None), context=self)
 
-        self.class_weight = loss_cls.class_weight
+        self.class_weight = loss_cls.get('class_weight', None)
         self.loss_cls = build_loss(loss_cls)
         self.loss_mask = build_loss(loss_mask)
         self.loss_dice = build_loss(loss_dice)

mmdet/models/detectors/maskformer.py

@@ -1,4 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import copy
+
 import mmcv
 import numpy as np
 
@@ -27,12 +29,12 @@ def __init__(self,
         if neck is not None:
             self.neck = build_neck(neck)
 
-        panoptic_head_ = panoptic_head.deepcopy()
+        panoptic_head_ = copy.deepcopy(panoptic_head)
         panoptic_head_.update(train_cfg=train_cfg)
         panoptic_head_.update(test_cfg=test_cfg)
         self.panoptic_head = build_head(panoptic_head_)
 
-        panoptic_fusion_head_ = panoptic_fusion_head.deepcopy()
+        panoptic_fusion_head_ = copy.deepcopy(panoptic_fusion_head)
         panoptic_fusion_head_.update(test_cfg=test_cfg)
         self.panoptic_fusion_head = build_head(panoptic_fusion_head_)