ml.postprocessing.edgeimpulse --- Edge Impulse

ml.postprocessing.edgeimpulse 模組包含用於 Edge Impulse 模型的後處理類別。

class Fomo -- Fast Objects More Objects

FOMO(Fast Objects More Objects)模型輸出的後處理器。

class ml.postprocessing.edgeimpulse.Fomo(threshold: float = 0.4, w_scale: float = 1.414214, h_scale: float = 1.414214, nms_threshold: float = 0.1, nms_sigma: float = 0.001)

建立一個 FOMO 後處理器。

threshold 保留一個偵測所需的最低分數。

w_scale 在非極大值抑制之前套用於網格儲存格寬度的水平縮放係數。較大的值會使相鄰儲存格合併為單一偵測。

h_scale 在非極大值抑制之前套用於網格儲存格高度的垂直縮放係數。較大的值會使相鄰儲存格合併為單一偵測。

nms_threshold 傳遞給非極大值抑制的 IoU 閾值。

nms_sigma 傳遞給非極大值抑制的 sigma 值(soft-NMS)。

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

ml.Model.predict() 以模型、其輸入及其原始輸出叫用。回傳一個依類別劃分的偵測清單。每個偵測是一個 ((x, y, w, h), score) 元組。空的類別清單會被包含在內,使得每個清單在輸出中的位置與模型輸出中的類別索引相符。當未偵測到任何東西時,回傳一個空元組。