v1.3.1 commit of bc75c9d6c

ca9829c1 · limm · e4fb2aa4 · ca9829c1
Commit ca9829c1 authored Jun 25, 2025 by limm
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdeploy/codebase/mmdet/models/dense_heads/base_dense_head.py ...ploy/codebase/mmdet/models/dense_heads/base_dense_head.py +1 -1

No files found.
--- a/mmdeploy/codebase/mmdet/models/dense_heads/base_dense_head.py
+++ b/mmdeploy/codebase/mmdet/models/dense_heads/base_dense_head.py
@@ -113,7 +113,7 @@ def base_dense_head__predict_by_feat(
        if self.use_sigmoid_cls:
            scores = scores.sigmoid()
        else:
-            scores = scores.softmax(-1)
+            scores = scores.softmax(-1)[:, :, :-1]
        if with_score_factors:
            score_factors = score_factors.permute(0, 2, 3,
                                                  1).reshape(batch_size,