Package org.jetbrains.kotlinx.dl.api.core.util

Types

TF

class TF

Helper class that emulates Keras functions from tensorflow.keras.

Functions

loadImageNetClassLabels

fun loadImageNetClassLabels(): MutableMap<Int, String>

Forms mapping of class label to class name for the ImageNet dataset.

predictTopNLabels

fun predictTopNLabels(it: InferenceModel, floatArray: FloatArray, labels: MutableMap<Int, String>, topN: Int = 5): MutableMap<Int, Pair<String, Float>>

Returns top-N labels for the given floatArray encoded with mapping labels.

serializeLabelsToBuffer

fun serializeLabelsToBuffer(src: FloatArray, amountOfClasses: Long): FloatBuffer

Converts src to FloatBuffer.

serializeToBuffer

fun serializeToBuffer(src: Array<FloatArray>): FloatBuffer

fun serializeToBuffer(src: FloatArray): FloatBuffer

Converts src to FloatBuffer.

fun serializeToBuffer(src: Array<FloatArray>, start: Int, length: Int): FloatBuffer

Converts src to FloatBuffer from start position for the next length positions.