Skip to content

GitLab

Unverified Commit 18cfeadb authored by Duino's avatar Duino Committed by GitHub
Browse files

Fix issue#219 (#221) 


Co-authored-by: default avatarbingwen.ai <bingwen.ai@bytedance.com>
parent 7142c933
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
segmentation/mmseg_custom/models/decode_heads/mask2former_head.py
View file @ 18cfeadb
...@@ -29,6 +29,8 @@ class Mask2FormerHead(BaseDecodeHead): ...@@ -29,6 +29,8 @@ class Mask2FormerHead(BaseDecodeHead):
feat_channels (int): Number of channels for features. feat_channels (int): Number of channels for features.
out_channels (int): Number of channels for output. out_channels (int): Number of channels for output.
num_classes (int): Number of classes. num_classes (int): Number of classes.
num_things_classes (int): Number of things.
num_stuff_classes (int): Number of stuff.
num_queries (int): Number of query in Transformer decoder. num_queries (int): Number of query in Transformer decoder.
pixel_decoder (:obj:`mmcv.ConfigDict` | dict): Config for pixel pixel_decoder (:obj:`mmcv.ConfigDict` | dict): Config for pixel
decoder. Defaults to None. decoder. Defaults to None.
...@@ -58,6 +60,8 @@ class Mask2FormerHead(BaseDecodeHead): ...@@ -58,6 +60,8 @@ class Mask2FormerHead(BaseDecodeHead):
feat_channels, feat_channels,
out_channels, out_channels,
num_classes=80, num_classes=80,
num_things_classes=None,
num_stuff_classes=None,
num_queries=100, num_queries=100,
num_transformer_feat_level=3, num_transformer_feat_level=3,
pixel_decoder=None, pixel_decoder=None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment