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 비최대 억제(soft-NMS)에 전달되는 시그마 값입니다.

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

모델, 그 입력, 그리고 원시 출력과 함께 ml.Model.predict() 에 의해 호출됩니다. 클래스별 검출 목록의 목록을 반환합니다. 각 검출은 ((x, y, w, h), score) 튜플입니다. 출력에서 각 목록의 위치가 모델 출력의 클래스 인덱스와 일치하도록 빈 클래스 목록도 포함됩니다. 아무것도 검출되지 않으면 빈 튜플을 반환합니다.