Revert support for newest numpy versions.

This commit is contained in:
ccarrizosa 2024-11-10 15:54:01 +01:00
parent 68351d2224
commit 2d8a7b67f8

View File

@ -242,7 +242,7 @@ def format_single_result(
)
class_name = class_names[np.argmax(class_overall)]
annotations = get_annotations_from_preds(predictions, class_names)
except (np.exceptions.AxisError, ValueError):
except (np.AxisError, ValueError):
# No detections
class_overall = np.zeros(len(class_names))
class_name = "None"