Update head.

mmdet/models/anchor_heads/decoupled_solo_head.py

@@ -441,7 +441,7 @@ class DecoupledSOLOHead(nn.Module):
         cate_scores = cate_scores[keep]
         sum_masks = sum_masks[keep]
         cate_labels = cate_labels[keep]
-        # mask scoring
+        # maskness
         seg_score = (seg_masks_soft * seg_masks.float()).sum((1, 2)) / sum_masks
         cate_scores *= seg_score
 

mmdet/models/anchor_heads/decoupled_solo_light_head.py

@@ -436,7 +436,7 @@ class DecoupledSOLOLightHead(nn.Module):
         cate_scores = cate_scores[keep]
         sum_masks = sum_masks[keep]
         cate_labels = cate_labels[keep]
-        # mask scoring
+        # maskness
         seg_score = (seg_masks_soft * seg_masks.float()).sum((1, 2)) / sum_masks
         cate_scores *= seg_score
 

mmdet/models/anchor_heads/solo_head.py

@@ -389,7 +389,7 @@ class SOLOHead(nn.Module):
         cate_scores = cate_scores[keep]
         cate_labels = cate_labels[keep]
 
-        # mask scoring.
+        # maskness.
         seg_scores = (seg_preds * seg_masks.float()).sum((1, 2)) / sum_masks
         cate_scores *= seg_scores
 

mmdet/models/anchor_heads/solov2_head.py

@@ -439,7 +439,7 @@ class SOLOv2Head(nn.Module):
         cate_scores = cate_scores[keep]
         cate_labels = cate_labels[keep]
 
-        # mask scoring.
+        # maskness.
         seg_scores = (seg_preds * seg_masks.float()).sum((1, 2)) / sum_masks
         cate_scores *= seg_scores
 

mmdet/models/anchor_heads/solov2_light_head.py

@@ -438,7 +438,7 @@ class SOLOv2LightHead(nn.Module):
         cate_scores = cate_scores[keep]
         cate_labels = cate_labels[keep]
 
-        # mask scoring.
+        # maskness.
         seg_scores = (seg_preds * seg_masks.float()).sum((1, 2)) / sum_masks
         cate_scores *= seg_scores