HyperEfficientNet

[原始碼]

HyperEfficientNet 類別

keras_tuner.applications.HyperEfficientNet(
    input_shape=None, input_tensor=None, classes=None, augmentation_model=None, **kwargs
)

一個 EfficientNet 超模型。

由 HyperEfficientNet 建構的模型會將形狀為 (高度、寬度、通道數) 的影像作為輸入。輸出為獨熱編碼，其長度與 classes 參數指定的類別數相符。

參數

• input_shape：可選的形狀元組，例如 (256, 256, 3)。必須指定 input_shape 或 input_tensor 其中之一。
• input_tensor：可選的 Keras 張量（即 layers.Input() 的輸出），用作模型的影像輸入。必須指定 input_shape 或 input_tensor 其中之一。
• classes：可選的影像分類類別數量，僅當 include_top 為 True 且未指定 weights 參數時才需要指定。
• augmentation_model：可選的用於影像增強的 Model 或 HyperModel 實例。
• **kwargs：適用於所有超模型的其他關鍵字參數。請參閱 keras_tuner.HyperModel。