backprop.models.efficientnet

backprop.models.efficientnet.model

class EfficientNet(model_path: str = 'efficientnet-b0', init_model=None, name: Optional[str] = None, description: Optional[str] = None, tasks: Optional[List[str]] = None, details: Optional[Dict] = None, device=None)

Bases: backprop.models.generic_models.PathModel

EfficientNet is a very efficient image-classification model. Trained on ImageNet.

model_path

Any efficientnet model (smaller to bigger) from efficientnet-b0 to efficientnet-b7

init_model

Callable that initialises the model from the model_path

name

string identifier for the model. Lowercase letters and numbers. No spaces/special characters except dashes.

description

String description of the model.

tasks

List of supported task strings

details

Dictionary of additional details about the model

device

Device for model. Defaults to “cuda” if available.

__call__(task_input, task='image-classification')

Uses the model for the image-classification task

Parameters

• task_input – input dictionary according to the image-classification task specification

• task – image-classification

configure_optimizers()

image_classification(image, top_k=10000)

static list_models()

pre_finetuning(labels=None, num_classes=None)

process_batch(params, task='image-classification')

training: bool

training_step(batch, task='image-classification')

backprop.models.efficientnet.models_list