diff --git a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModels.kt b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModels.kt index d60f8d96e..d2553bf9b 100644 --- a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModels.kt +++ b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModels.kt @@ -574,6 +574,10 @@ public interface ModelType { */ public val channelsFirst: Boolean + /** + * An expected channels order for the input image. + * Note: the wrong choice of this parameter can significantly impact the model's performance. + */ public val inputColorMode: ColorMode /**