Device fixing #4

This commit is contained in:
mbsantiago 2025-08-25 23:11:33 +01:00
parent 9d4a9fc35c
commit 67e37227f5

View File

@ -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()]