CV

sealed class CV<T : GraphTrainableModel> : TFModelType<T, ImageRecognitionModel>

Image recognition models and preprocessing.

Types

DenseNet121

class DenseNet121(inputShape: IntArray?) : TFModels.CV<Functional>

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

DenseNet169

class DenseNet169(inputShape: IntArray?) : TFModels.CV<Functional>

DenseNet201

class DenseNet201(inputShape: IntArray?) : TFModels.CV<Functional>

Inception

class Inception(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

MobileNet

class MobileNet(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

MobileNetV2

class MobileNetV2(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

NASNetLarge

class NASNetLarge(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

NASNetMobile

class NASNetMobile(noTop: Boolean) : TFModels.CV<Functional>

ResNet101

class ResNet101(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

ResNet101v2

class ResNet101v2(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

ResNet152

class ResNet152(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

ResNet152v2

class ResNet152v2(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

ResNet18

class ResNet18(inputShape: IntArray?) : TFModels.CV<Functional>

ResNet34

class ResNet34(inputShape: IntArray?) : TFModels.CV<Functional>

ResNet50

class ResNet50(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

ResNet50v2

class ResNet50v2(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

VGG16

class VGG16(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Sequential>

VGG19

class VGG19(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Sequential>

Xception

class Xception(noTop: Boolean, inputShape: IntArray?) : TFModels.CV<Functional>

Functions

loadModelConfiguration

open override fun loadModelConfiguration(jsonFile: File): T

Loads model configuration from the provided jsonFile.

model

open fun model(modelHub: ModelHub): T

pretrainedModel

open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel

Properties

inputShape

var inputShape: IntArray? = null

modelRelativePath

open override val modelRelativePath: String

preprocessor

open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Inheritors

TFModels.CV