mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #2333 from MengzhangLI/fix_knet_master

Browse Source 
[Fix] Fix KNet IterativeDecodeHead bug in master branch
This commit is contained in:
Miao Zheng 2022-12-01 11:05:19 +08:00 committed by GitHub
commit 77dbecd5f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmseg/models/decode_heads/knet_head.py
View File

@ -411,6 +411,9 @@ class IterativeDecodeHead(BaseDecodeHead):
def __init__(self, num_stages, kernel_generate_head, kernel_update_head, def __init__(self, num_stages, kernel_generate_head, kernel_update_head,
**kwargs): **kwargs):
# ``IterativeDecodeHead`` would skip initialization of
# ``BaseDecodeHead`` which would be called when building
# ``self.kernel_generate_head``.
super(BaseDecodeHead, self).__init__(**kwargs) super(BaseDecodeHead, self).__init__(**kwargs)
assert num_stages == len(kernel_update_head) assert num_stages == len(kernel_update_head)
self.num_stages = num_stages self.num_stages = num_stages
@ -420,6 +423,7 @@ class IterativeDecodeHead(BaseDecodeHead):
self.num_classes = self.kernel_generate_head.num_classes self.num_classes = self.kernel_generate_head.num_classes
self.input_transform = self.kernel_generate_head.input_transform self.input_transform = self.kernel_generate_head.input_transform
self.ignore_index = self.kernel_generate_head.ignore_index self.ignore_index = self.kernel_generate_head.ignore_index
self.out_channels = self.num_classes
for head_cfg in kernel_update_head: for head_cfg in kernel_update_head:
self.kernel_update_head.append(build_head(head_cfg)) self.kernel_update_head.append(build_head(head_cfg))