ml.postprocessing.ultralytics --- Ultralytics

Module ml.postprocessing.ultralytics chứa các lớp cho Ultralytics.

class YoloV5 -- YOLO V5

Được dùng để xử lý hậu kỳ đầu ra của mô hình YOLO V5.

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

Tạo một bộ xử lý hậu kỳ YOLO V5.

threshold Ngưỡng điểm số để lọc các phát hiện trước khi NMS.

nms_threshold Ngưỡng IoU được dùng bởi NMS để loại bỏ các hộp giới hạn chồng lấp.

nms_sigma Giá trị Sigma được dùng bởi Soft-NMS để phân rã điểm số.

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

Xử lý hậu kỳ danh sách outputs của các mảng được tạo ra bởi model cho danh sách inputs đã cho. Trả về danh sách các danh sách theo từng lớp; mỗi danh sách bên trong chứa các tuple ((x, y, w, h), score) theo tọa độ ROI. Các danh sách lớp rỗng được giữ lại để mỗi chỉ số danh sách khớp với chỉ số lớp tương ứng trong đầu ra của mô hình.

class YoloV8 -- YOLO V8

Được dùng để xử lý hậu kỳ đầu ra của mô hình YOLO V8.

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

Tạo một bộ xử lý hậu kỳ YOLO V8.

threshold Ngưỡng điểm số để lọc các phát hiện trước khi NMS.

nms_threshold Ngưỡng IoU được dùng bởi NMS để loại bỏ các hộp giới hạn chồng lấp.

nms_sigma Giá trị Sigma được dùng bởi Soft-NMS để phân rã điểm số.

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

Xử lý hậu kỳ danh sách outputs của các mảng được tạo ra bởi model cho danh sách inputs đã cho. Trả về danh sách các danh sách theo từng lớp; mỗi danh sách bên trong chứa các tuple ((x, y, w, h), score) theo tọa độ ROI. Các danh sách lớp rỗng được giữ lại để mỗi chỉ số danh sách khớp với chỉ số lớp tương ứng trong đầu ra của mô hình.