Commit 3f8604b
authored
fix(mmdet/base_dense_head): 🐛 correct slicing issue in score softmax for single-class models (open-mmlab#2827)
The softmax function previously sliced out the background class, which
caused an IndexError in single-class model export. This update removes
the slicing of background class scores, fixing the issue when exporting
RetinaNet with single-class configurations.
- Updated `base_dense_head__predict_by_feat` to prevent IndexError when
handling single-class outputs.
- This change is particularly relevant for RetinaNet model exports.1 parent bc75c9d commit 3f8604b
File tree
1 file changed
+1
-1
lines changed- mmdeploy/codebase/mmdet/models/dense_heads
1 file changed
+1
-1
lines changedLines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
113 | 113 | | |
114 | 114 | | |
115 | 115 | | |
116 | | - | |
| 116 | + | |
117 | 117 | | |
118 | 118 | | |
119 | 119 | | |
| |||
0 commit comments