From 589d8ac224f43f57403c0ebf99326d325731244f Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 12:40:19 +0800 Subject: [PATCH 1/2] [Fix] Fix KNet IterativeDecodeHead bug in v0.x branch --- mmseg/models/decode_heads/knet_head.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmseg/models/decode_heads/knet_head.py b/mmseg/models/decode_heads/knet_head.py index f73daccb6..bb5dec590 100644 --- a/mmseg/models/decode_heads/knet_head.py +++ b/mmseg/models/decode_heads/knet_head.py @@ -420,6 +420,7 @@ class IterativeDecodeHead(BaseDecodeHead): self.num_classes = self.kernel_generate_head.num_classes self.input_transform = self.kernel_generate_head.input_transform self.ignore_index = self.kernel_generate_head.ignore_index + self.out_channels = self.num_classes for head_cfg in kernel_update_head: self.kernel_update_head.append(build_head(head_cfg)) From 91440718dee357612b480acfae9bae265dfe4ea8 Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 22:38:10 +0800 Subject: [PATCH 2/2] add comment --- mmseg/models/decode_heads/knet_head.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mmseg/models/decode_heads/knet_head.py b/mmseg/models/decode_heads/knet_head.py index bb5dec590..78a270277 100644 --- a/mmseg/models/decode_heads/knet_head.py +++ b/mmseg/models/decode_heads/knet_head.py @@ -411,6 +411,9 @@ class IterativeDecodeHead(BaseDecodeHead): def __init__(self, num_stages, kernel_generate_head, kernel_update_head, **kwargs): + # ``IterativeDecodeHead`` would skip initialization of + # ``BaseDecodeHead`` which would be called when building + # ``self.kernel_generate_head``. super(BaseDecodeHead, self).__init__(**kwargs) assert num_stages == len(kernel_update_head) self.num_stages = num_stages