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

Types

TF
Link copied to clipboard
class TF
Content copied to clipboard

Helper class that emulates Keras functions from tensorflow.keras.

Functions

loadImageNetClassLabels
Link copied to clipboard
fun loadImageNetClassLabels(): MutableMap<Int, String>
Content copied to clipboard

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

mapOfNotNull
Link copied to clipboard
fun <K, V> mapOfNotNull(vararg mapping: Pair<K?, V?>): Map<K, V>
Content copied to clipboard

Create a new read-only map from a list of pairs, if both values in the pair are not null.

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

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

serializeLabelsToBuffer
Link copied to clipboard
fun serializeLabelsToBuffer(src: FloatArray, amountOfClasses: Long): FloatBuffer
Content copied to clipboard

Converts src to FloatBuffer.

serializeToBuffer
Link copied to clipboard
fun serializeToBuffer(src: Array<FloatArray>): FloatBuffer
Content copied to clipboard
fun serializeToBuffer(src: FloatArray): FloatBuffer
Content copied to clipboard

Converts src to FloatBuffer.

fun serializeToBuffer(src: Array<FloatArray>, start: Int, length: Int): FloatBuffer
Content copied to clipboard

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