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

Device fixing #2

Browse Source
This commit is contained in:
mbsantiago 2025-08-25 23:07:09 +01:00
parent a267db290c
commit d0bab60bf3
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/batdetect2/postprocess/extraction.py
View File

@ -47,8 +47,8 @@ def extract_prediction_tensor(
indexing="ij",
)
freqs = freqs.flatten()
times = times.flatten()
freqs = freqs.flatten().to(detection_heatmap)
times = times.flatten().to(detection_heatmap)
output_size_preds = output.size_preds.detach()
output_features = output.features.detach()
@ -58,7 +58,6 @@ def extract_prediction_tensor(
for idx, item in enumerate(detection_heatmap):
item = item.squeeze().flatten() # Remove channel dim
indices = torch.argsort(item, descending=True)[:max_detections]
indices.to(detection_heatmap)
detection_scores = item.take(indices)
detection_freqs = freqs.take(indices)

3
src/batdetect2/train/labels.py
View File

@ -210,6 +210,9 @@ def generate_heatmaps(
indexing="ij",
)
freqs = freqs.to(spec)
times = times.to(spec)
for sound_event_annotation in clip_annotation.sound_events:
geom = sound_event_annotation.sound_event.geometry
if geom is None: