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 = [] 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()]