lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-10 17:19:34 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Small fix

Browse Source
This commit is contained in:
mbsantiago 2025-08-18 10:25:14 +01:00
parent 9d1497b3f4
commit 98f83e8b34
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/batdetect2/models/detectors.py
View File

@ -114,8 +114,8 @@ class Detector(DetectionModel):
`(B, C_out, H, W)`. `(B, C_out, H, W)`.
""" """
features = self.backbone(spec) features = self.backbone(spec)
detection = self.detector_head(features)
classification = self.classifier_head(features) classification = self.classifier_head(features)
detection = classification.sum(dim=1, keep_dim=True)
size_preds = self.bbox_head(features) size_preds = self.bbox_head(features)
return ModelOutput( return ModelOutput(
detection_probs=detection, detection_probs=detection,