mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-01-10 17:19:34 +01:00
Device fixing #4
This commit is contained in:
parent
9d4a9fc35c
commit
67e37227f5
@ -34,12 +34,12 @@ def convert_detections_to_raw_predictions(
|
||||
predictions = []
|
||||
|
||||
for score, class_scores, time, freq, dims, feats in zip(
|
||||
detections.scores,
|
||||
detections.class_scores,
|
||||
detections.times,
|
||||
detections.frequencies,
|
||||
detections.sizes,
|
||||
detections.features,
|
||||
detections.scores.cpu().numpy(),
|
||||
detections.class_scores.cpu().numpy(),
|
||||
detections.times.cpu().numpy(),
|
||||
detections.frequencies.cpu().numpy(),
|
||||
detections.sizes.cpu().numpy(),
|
||||
detections.features.cpu().numpy(),
|
||||
):
|
||||
highest_scoring_class = targets.class_names[class_scores.argmax()]
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user