Package org.jetbrains.kotlinx.dl.api.inference.loaders

Types

TFModelHub
Link copied to clipboard
class TFModelHub(cacheDirectory: File) : ModelHub

This class provides methods for loading Keras models, its weights and ImageNet labels (for prediction purposes) to the local cacheDirectory.

TFModels
Link copied to clipboard
object TFModels

Supported models for inference and transfer learning, trained on ImageNet dataset.

TFModelType
Link copied to clipboard
interface TFModelType<T : TensorFlowInferenceModel, U : InferenceModel> : ModelType<T, U>

Base ModelType for TensorFlow models.