EfficientNet

biapy.models.efficientnet.efficientnet(efficientnet_name, image_shape, n_classes=2, load_imagenet_weights=True)[source]

Create EfficientNet.

Reference: EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks.

Parameters:

  • efficientnet_name (str, optional) – Efficientnet model name to be loaded. Available options: “efficientnet_b[0-7]”

  • image_shape (2D tuple) – Dimensions of the input image.

  • n_classes (int, optional) – Number of classes.

Returns:

model – EfficientNet model.

Return type:

Torch model