From ee025ea33bd34307c83bd321c020b1722bed326d Mon Sep 17 00:00:00 2001 From: MeowZheng Date: Wed, 8 Jun 2022 13:05:41 +0800 Subject: [PATCH] [Fix]Add build head --- mmseg/models/decode_heads/knet_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/models/decode_heads/knet_head.py b/mmseg/models/decode_heads/knet_head.py index 46de21434..74f371598 100644 --- 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