output_type module

Classification

class edgeimpulse.model.output_type.Classification(
		labels: Optional[List[str]] = None
)

Parameters

  • labels: Optional[List[str]] = None

Bases

  • builtins.dict

ObjectDetection

class edgeimpulse.model.output_type.ObjectDetection(
		labels: List[str],
		last_layer: Union[Literal['mobilenet-ssd''fomo''yolov5''yolo5v5-drpai''yolox''yolov7'], str],
		minimum_confidence: float
)

Parameters

  • labels: List[str]

  • last_layer: Union[Literal['mobilenet-ssd', 'fomo', 'yolov5', 'yolo5v5-drpai', 'yolox', 'yolov7'], str]

  • minimum_confidence: float

Bases

  • builtins.dict

Regression

class edgeimpulse.model.output_type.Regression(
		
)

Bases

  • builtins.dict

Last updated