[Fix]Add build head

2022-06-08 13:05:41 +08:00 · 2022-06-08 13:05:41 +08:00 · ee025ea33b
parent 49ada036d8
commit ee025ea33b
1 changed files with 1 additions and 1 deletions
--- a/mmseg/models/decode_heads/knet_head.py
+++ b/mmseg/models/decode_heads/knet_head.py
@ -414,7 +414,7 @@ class IterativeDecodeHead(BaseDecodeHead):
        super(BaseDecodeHead, self).__init__(**kwargs)
        assert num_stages == len(kernel_update_head)
        self.num_stages = num_stages
-        self.kernel_generate_head = (kernel_generate_head)
+        self.kernel_generate_head = MODELS.build(kernel_generate_head)
        self.kernel_update_head = nn.ModuleList()
        self.align_corners = self.kernel_generate_head.align_corners
        self.num_classes = self.kernel_generate_head.num_classes