ml.postprocessing.ultralytics — Ultralytics

Le module ml.postprocessing.ultralytics contient des classes pour Ultralytics.

class YoloV5 – YOLO V5

Utilisé pour post-traiter la sortie du modèle YOLO V5.

class ml.postprocessing.ultralytics.YoloV5(threshold: float = 0.6, nms_threshold: float = 0.1, nms_sigma: float = 0.1)

Crée un post-traitement YOLO V5.

threshold Seuil de score pour filtrer les détections avant la NMS.

nms_threshold Seuil d’IoU utilisé par la NMS pour supprimer les boîtes englobantes qui se chevauchent.

nms_sigma Valeur sigma utilisée par la Soft-NMS pour la décroissance des scores.

__call__(model: ml.Model, inputs: list, outputs: list) → list

Post-traite la liste de tableaux outputs produite par model pour la liste inputs donnée. Renvoie une liste de listes par classe ; chaque liste interne contient des tuples ((x, y, w, h), score) en coordonnées de ROI. Les listes de classes vides sont conservées afin que l’index de chaque liste corresponde à l’index de classe correspondant dans la sortie du modèle.

class YoloV8 – YOLO V8

Utilisé pour post-traiter la sortie du modèle YOLO V8.

class ml.postprocessing.ultralytics.YoloV8(threshold: float = 0.6, nms_threshold: float = 0.1, nms_sigma: float = 0.1)

Crée un post-traitement YOLO V8.

threshold Seuil de score pour filtrer les détections avant la NMS.

nms_threshold Seuil d’IoU utilisé par la NMS pour supprimer les boîtes englobantes qui se chevauchent.

nms_sigma Valeur sigma utilisée par la Soft-NMS pour la décroissance des scores.

__call__(model: ml.Model, inputs: list, outputs: list) → list

Post-traite la liste de tableaux outputs produite par model pour la liste inputs donnée. Renvoie une liste de listes par classe ; chaque liste interne contient des tuples ((x, y, w, h), score) en coordonnées de ROI. Les listes de classes vides sont conservées afin que l’index de chaque liste corresponde à l’index de classe correspondant dans la sortie du modèle.