ml.postprocessing.edgeimpulse — Edge Impulse

O módulo ml.postprocessing.edgeimpulse contém classes de pós-processamento para modelos Edge Impulse.

class Fomo – Fast Objects More Objects

Pós-processador para a saída do modelo 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)

Cria um pós-processador FOMO.

threshold pontuação mínima exigida para que uma deteção seja mantida.

w_scale fator de escala horizontal aplicado à largura da célula da grelha antes da supressão não-máxima. Valores maiores fazem com que células vizinhas sejam fundidas numa única deteção.

h_scale fator de escala vertical aplicado à altura da célula da grelha antes da supressão não-máxima. Valores maiores fazem com que células vizinhas sejam fundidas numa única deteção.

nms_threshold Limiar IoU passado à supressão não-máxima.

nms_sigma valor sigma passado à supressão não-máxima (soft-NMS).

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

Invocado por ml.Model.predict() com o modelo, as suas entradas e as suas saídas brutas. Devolve uma lista de listas de deteções por classe. Cada deteção é um tuplo ((x, y, w, h), score). As listas de classes vazias são incluídas para que a posição de cada lista na saída corresponda ao índice de classe na saída do modelo. Devolve um tuplo vazio quando nada é detetado.