CV

sealed class CV<T : GraphTrainableModel> : ModelType<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.api.core.util.loadImageNetClassLabels 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

model

open fun model(modelHub: ModelHub): T

Loads the model, identified by this name, from the modelHub.

preInit

open fun preInit(): InferenceModel

preprocessInput

open fun preprocessInput(imageFile: File, preprocessing: Preprocessing): FloatArray

abstract fun preprocessInput(data: FloatArray, tensorShape: LongArray): FloatArray

Common preprocessing function for the Neural Networks trained on ImageNet and whose weights are available with the keras.application.

pretrainedModel

open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel

Returns the specially prepared pre-trained model of the type U.

Properties

channelsFirst

open override val channelsFirst: Boolean = false

If true it means that the second dimension is related to number of channels in image has short notation as NCWH, otherwise, channels are at the last position and has a short notation as NHWC.

inputShape

var inputShape: IntArray? = null

modelRelativePath

open override val modelRelativePath: String

Relative path to model for local and S3 buckets storages.

Inheritors

TFModels.CV