BASNet Segmentation

• 원본 링크 : https://keras.io/api/keras_cv/models/tasks/basnet_segmentation/
• 최종 확인 : 2024-11-25

source

BASNet class

keras_cv.models.BASNet(
    backbone,
    num_classes,
    input_shape=(None, None, 3),
    input_tensor=None,
    include_rescaling=False,
    projection_filters=64,
    prediction_heads=None,
    refinement_head=None,
    **kwargs
)

A Keras model implementing the BASNet architecture for semantic segmentation.

References

• BASNet: Boundary-Aware Segmentation Network for Mobile and Web Applications

Arguments

• backbone: keras.Model. The backbone network for the model that is used as a feature extractor for BASNet prediction encoder. Currently supported backbones are ResNet18 and ResNet34. Default backbone is keras_cv.models.ResNet34Backbone() (Note: Do not specify ‘input_shape’, ‘input_tensor’, or ‘include_rescaling’ within the backbone. Please provide these while initializing the ‘BASNet’ model.)
• num_classes: int, the number of classes for the segmentation model.
• input_shape: optional shape tuple, defaults to (None, None, 3).
• input_tensor: optional Keras tensor (i.e., output of layers.Input()) to use as image input for the model.
• include_rescaling: bool, whether to rescale the inputs. If set to True, inputs will be passed through a Rescaling(1/255.0) layer.
• projection_filters: int, number of filters in the convolution layer projecting low-level features from the backbone.
• prediction_heads: (Optional) List of keras.layers.Layer defining the prediction module head for the model. If not provided, a default head is created with a Conv2D layer followed by resizing.
• refinement_head: (Optional) a keras.layers.Layer defining the refinement module head for the model. If not provided, a default head is created with a Conv2D layer.

Example

import keras_cv
images = np.ones(shape=(1, 288, 288, 3))
labels = np.zeros(shape=(1, 288, 288, 1))
# Note: Do not specify 'input_shape', 'input_tensor', or
# 'include_rescaling' within the backbone.
backbone = keras_cv.models.ResNet34Backbone()
model = keras_cv.models.segmentation.BASNet(
    backbone=backbone,
    num_classes=1,
    input_shape=[288, 288, 3],
    include_rescaling=False
)
# Evaluate model
output = model(images)
pred_labels = output[0]
# Train model
model.compile(
    optimizer="adam",
    loss=keras.losses.BinaryCrossentropy(from_logits=False),
    metrics=["accuracy"],
)
model.fit(images, labels, epochs=3)

source

from_preset method

BASNet.from_preset()

Instantiate BASNet model from preset config and weights.

Arguments

• preset: string. Must be one of “resnet18”, “resnet34”, “basnet_resnet18”, “basnet_resnet34”. If looking for a preset with pretrained weights, choose one of “”.
• load_weights: Whether to load pre-trained weights into model. Defaults to None, which follows whether the preset has pretrained weights available.
• input_shape: input shape that will be passed to backbone initialization, Defaults to None.If None, the preset value will be used.

Example

model = keras_cv.models.BASNet.from_preset(
  "",
)

model = keras_cv.models.BASNet.from_preset(
  "",
  load_weights=False,
)