DeepLabV3ImageSegmenter 模型

[来源]

DeepLabV3ImageSegmenter 类

keras_hub.models.DeepLabV3ImageSegmenter(
    backbone, num_classes, activation=None, preprocessor=None, **kwargs
)

DeepLabV3 和 DeeplabV3 以及 DeeplabV3Plus 分割任务。

参数

• backbone: keras_hub.models.DeepLabV3 实例。
• num_classes: int。检测模型的类别数。请注意，num_classes 包含背景类，数据中的类应由范围为 [0, num_classes] 的整数表示。
• activation: str 或 callable。Dense 层上使用的激活函数。将 activation=None 设置为返回输出 logits。默认为 None。
• preprocessor: keras_hub.models.DeepLabV3ImageSegmenterPreprocessor 或 None。如果为 None，此模型将不会应用预处理，并且输入应在调用模型之前进行预处理。

示例

加载具有所有 21 个类的 DeepLabV3 预设，预训练分割头部。

images = np.ones(shape=(1, 96, 96, 3))
labels = np.zeros(shape=(1, 96, 96, 1))
segmenter = keras_hub.models.DeepLabV3ImageSegmenter.from_preset(
    "deeplabv3_resnet50_pascalvoc",
)
segmenter.predict(images)

指定 num_classes 以加载随机初始化的分割头部。

segmenter = keras_hub.models.DeepLabV3ImageSegmenter.from_preset(
    "deeplabv3_resnet50_pascalvoc",
    num_classes=2,
)
segmenter.fit(images, labels, epochs=3)
segmenter.predict(images)  # Trained 2 class segmentation.

加载 DeepLabv3+ 预设，它是 DeepLabV3 的扩展，通过添加一个简单但有效的解码器模块来细化分割结果，尤其是在对象边界处。

segmenter = keras_hub.models.DeepLabV3ImageSegmenter.from_preset(
    "deeplabv3_plus_resnet50_pascalvoc",
)
segmenter.predict(images)

[来源]

from_preset 方法

DeepLabV3ImageSegmenter.from_preset(preset, load_weights=True, **kwargs)

从模型预设实例化 keras_hub.models.Task。

预设是用于保存和加载预训练模型的配置、权重和其他文件资产的目录。preset 可以作为以下之一传递：

1. 内置预设标识符，如 'bert_base_en'
2. Kaggle 模型句柄，如 'kaggle://user/bert/keras/bert_base_en'
3. Hugging Face 句柄，如 'hf://user/bert_base_en'
4. 本地预设目录的路径，如 './bert_base_en'

对于任何 Task 子类，您可以运行 cls.presets.keys() 来列出该类上可用的所有内置预设。

此构造函数可以通过两种方式之一调用。从特定任务的基类（如 keras_hub.models.CausalLM.from_preset()），或从模型类（如 keras_hub.models.BertTextClassifier.from_preset()）调用。如果从基类调用，返回对象的子类将从预设目录中的配置推断。

参数

• preset: string。内置预设标识符、Kaggle 模型句柄、Hugging Face 句柄或本地目录的路径。
• load_weights: bool。如果为 True，则保存的权重将加载到模型架构中。如果为 False，则所有权重都将被随机初始化。

示例

# Load a Gemma generative task.
causal_lm = keras_hub.models.CausalLM.from_preset(
    "gemma_2b_en",
)

# Load a Bert classification task.
model = keras_hub.models.TextClassifier.from_preset(
    "bert_base_en",
    num_classes=2,
)

backbone 属性

keras_hub.models.DeepLabV3ImageSegmenter.backbone

具有核心架构的 keras_hub.models.Backbone 模型。

preprocessor 属性

keras_hub.models.DeepLabV3ImageSegmenter.preprocessor

用于预处理输入的 keras_hub.models.Preprocessor 层。