Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix(mmdet/base_dense_head): 🐛 correct slicing issue in score softmax …
…for single-class models (#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.
- Loading branch information