ml.postprocessing.edgeimpulse — Edge Impulse

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

classe 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 detecção seja mantida.

w_scale fator de escala horizontal aplicado à largura da célula da grade antes da supressão não máxima. Valores maiores fazem com que células vizinhas sejam mescladas em uma única detecção.

h_scale fator de escala vertical aplicado à altura da célula da grade antes da supressão não máxima. Valores maiores fazem com que células vizinhas sejam mescladas em uma única detecção.

nms_threshold limiar de IoU passado para a supressão não máxima.

nms_sigma valor sigma passado para a 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, suas entradas e suas saídas brutas. Retorna uma lista de listas de detecção por classe. Cada detecção é uma tupla ((x, y, w, h), score). Listas de classe 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. Retorna uma tupla vazia quando nada é detectado.