ml.postprocessing.ultralytics — Ultralytics

ml.postprocessing.ultralytics-moduuli sisältää luokkia Ultralyticsia varten.

luokka YoloV5 – YOLO V5

Käytetään YOLO V5 -mallin ulostulon jälkikäsittelyyn.

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

Luo YOLO V5 -jälkikäsittelijän.

threshold Pistemäärän kynnysarvo tunnistusten suodattamiseen ennen NMS:ää.

nms_threshold IoU-kynnysarvo, jota NMS käyttää päällekkäisten rajauslaatikoiden vaimentamiseen.

nms_sigma Sigma-arvo, jota Soft-NMS käyttää pistemäärän vaimenemiseen.

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

Jälkikäsittelee model-mallin annetulle inputs-luettelolle tuottaman taulukoiden outputs-luettelon. Palauttaa luettelon luokkakohtaisista luetteloista; kukin sisäluettelo sisältää ((x, y, w, h), score) -monikoita ROI-koordinaateissa. Tyhjät luokkaluettelot säilytetään, jotta kunkin luettelon indeksi vastaa mallin ulostulon vastaavaa luokkaindeksiä.

luokka YoloV8 – YOLO V8

Käytetään YOLO V8 -mallin ulostulon jälkikäsittelyyn.

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

Luo YOLO V8 -jälkikäsittelijän.

threshold Pistemäärän kynnysarvo tunnistusten suodattamiseen ennen NMS:ää.

nms_threshold IoU-kynnysarvo, jota NMS käyttää päällekkäisten rajauslaatikoiden vaimentamiseen.

nms_sigma Sigma-arvo, jota Soft-NMS käyttää pistemäärän vaimenemiseen.

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

Jälkikäsittelee model-mallin annetulle inputs-luettelolle tuottaman taulukoiden outputs-luettelon. Palauttaa luettelon luokkakohtaisista luetteloista; kukin sisäluettelo sisältää ((x, y, w, h), score) -monikoita ROI-koordinaateissa. Tyhjät luokkaluettelot säilytetään, jotta kunkin luettelon indeksi vastaa mallin ulostulon vastaavaa luokkaindeksiä.