ml.postprocessing.ultralytics --- Ultralytics

Modul ml.postprocessing.ultralytics berisi kelas untuk Ultralytics.

class YoloV5 -- YOLO V5

Digunakan untuk pasca-pemrosesan output model YOLO V5.

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

Membuat pasca-prosesor YOLO V5.

threshold Ambang batas skor untuk memfilter deteksi sebelum NMS.

nms_threshold Ambang batas IoU yang digunakan oleh NMS untuk menekan kotak pembatas yang tumpang tindih.

nms_sigma Nilai sigma yang digunakan oleh Soft-NMS untuk penurunan skor.

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

Pasca-pemrosesan daftar outputs array yang dihasilkan oleh model untuk daftar inputs yang diberikan. Mengembalikan daftar daftar per-kelas; setiap daftar dalam berisi tuple ((x, y, w, h), score) dalam koordinat ROI. Daftar kelas yang kosong tetap disertakan agar setiap indeks daftar sesuai dengan indeks kelas yang sesuai dalam output model.

class YoloV8 -- YOLO V8

Digunakan untuk pasca-pemrosesan output model YOLO V8.

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

Membuat pasca-prosesor YOLO V8.

threshold Ambang batas skor untuk memfilter deteksi sebelum NMS.

nms_threshold Ambang batas IoU yang digunakan oleh NMS untuk menekan kotak pembatas yang tumpang tindih.

nms_sigma Nilai sigma yang digunakan oleh Soft-NMS untuk penurunan skor.

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

Pasca-pemrosesan daftar outputs array yang dihasilkan oleh model untuk daftar inputs yang diberikan. Mengembalikan daftar daftar per-kelas; setiap daftar dalam berisi tuple ((x, y, w, h), score) dalam koordinat ROI. Daftar kelas yang kosong tetap disertakan agar setiap indeks daftar sesuai dengan indeks kelas yang sesuai dalam output model.