ml.postprocessing.ultralytics — Ultralytics

Modul ml.postprocessing.ultralytics sadrži klase za Ultralytics.

klasa YoloV5 – YOLO V5

Koristi se za naknadnu obradu izlaza YOLO V5 modela.

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

Stvara YOLO V5 naknadni obrađivač.

threshold Prag rezultata za filtriranje detekcija prije NMS-a.

nms_threshold IoU prag koji NMS koristi za potiskivanje preklapajućih graničnih okvira.

nms_sigma Sigma vrijednost koju Soft-NMS koristi za opadanje rezultata.

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

Naknadno obrađuje popis nizova outputs koji je model proizveo za zadani popis inputs. Vraća popis popisa po klasi; svaki unutarnji popis sadrži n-torke ((x, y, w, h), score) u ROI koordinatama. Prazni popisi klasa čuvaju se kako bi svaki indeks popisa odgovarao odgovarajućem indeksu klase u izlazu modela.

klasa YoloV8 – YOLO V8

Koristi se za naknadnu obradu izlaza YOLO V8 modela.

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

Stvara YOLO V8 naknadni obrađivač.

threshold Prag rezultata za filtriranje detekcija prije NMS-a.

nms_threshold IoU prag koji NMS koristi za potiskivanje preklapajućih graničnih okvira.

nms_sigma Sigma vrijednost koju Soft-NMS koristi za opadanje rezultata.

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

Naknadno obrađuje popis nizova outputs koji je model proizveo za zadani popis inputs. Vraća popis popisa po klasi; svaki unutarnji popis sadrži n-torke ((x, y, w, h), score) u ROI koordinatama. Prazni popisi klasa čuvaju se kako bi svaki indeks popisa odgovarao odgovarajućem indeksu klase u izlazu modela.