imlab.train_classifier.train

imlab.train_classifier.train(datas: dict, model_type: str = 'MobileNetV2', image_size: int = 300, checkpoint: str = '', add: bool = False) → Classifier

train.

Parameters

• datas (dict) – {label:[image]}

• model_type (str) –

• image_size (int) –

• checkpoint (str) –

• add (bool) –

Return type

Classifier