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 = []
|
predictions = []
|
||||||
|
|
||||||
for score, class_scores, time, freq, dims, feats in zip(
|
for score, class_scores, time, freq, dims, feats in zip(
|
||||||
detections.scores,
|
detections.scores.cpu().numpy(),
|
||||||
detections.class_scores,
|
detections.class_scores.cpu().numpy(),
|
||||||
detections.times,
|
detections.times.cpu().numpy(),
|
||||||
detections.frequencies,
|
detections.frequencies.cpu().numpy(),
|
||||||
detections.sizes,
|
detections.sizes.cpu().numpy(),
|
||||||
detections.features,
|
detections.features.cpu().numpy(),
|
||||||
):
|
):
|
||||||
highest_scoring_class = targets.class_names[class_scores.argmax()]
|
highest_scoring_class = targets.class_names[class_scores.argmax()]
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user