From 58e73c3fab714c6ee7d1e7c02633f9dffa4550a5 Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 12:46:24 +0800 Subject: [PATCH 1/3] [Fix] Fix KNet IterativeDecodeHead bug in dev-1.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 181ae0216..ae7afc96b 100644 --- a/mmseg/models/decode_heads/knet_head.py +++ b/mmseg/models/decode_heads/knet_head.py @@ -422,6 +422,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(MODELS.build(head_cfg)) From 8b618207514ff75e624dab94f10fe45609c87c6c Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 22:42:11 +0800 Subject: [PATCH 2/3] add comment --- data | 1 + mmseg/models/decode_heads/knet_head.py | 3 +++ 2 files changed, 4 insertions(+) create mode 120000 data diff --git a/data b/data new file mode 120000 index 000000000..af1f36392 --- /dev/null +++ b/data @@ -0,0 +1 @@ +/home/PJLAB/limengzhang/Downloads/datasets/ \ No newline at end of file diff --git a/mmseg/models/decode_heads/knet_head.py b/mmseg/models/decode_heads/knet_head.py index ae7afc96b..82d3a2807 100644 --- a/mmseg/models/decode_heads/knet_head.py +++ b/mmseg/models/decode_heads/knet_head.py @@ -413,6 +413,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 From 25abd9e92b097077e006dccd1b662f58cf95b228 Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 22:44:56 +0800 Subject: [PATCH 3/3] delete data link --- data | 1 - 1 file changed, 1 deletion(-) delete mode 120000 data diff --git a/data b/data deleted file mode 120000 index af1f36392..000000000 --- a/data +++ /dev/null @@ -1 +0,0 @@ -/home/PJLAB/limengzhang/Downloads/datasets/ \ No newline at end of file