tool/io.github.kmplitert.tool.expand

Package-level declarations

Properties

dimensions

val LiteRTTensorType.dimensions: List<Int>

Returns the dimensions of the tensor, or an empty list if not available.

totalElements

val LiteRTLayout.totalElements: Int

Returns the total number of elements in the tensor layout.

Functions

argmax

fun FloatArray.argmax(): Int

Returns the index of the maximum value in the FloatArray.

calculateIou

fun calculateIou(box1: FloatArray, box2: FloatArray): Float

Calculates the Intersection over Union (IoU) of two bounding boxes.

clamp

fun FloatArray.clamp(min: Float, max: Float): FloatArray

Clamps the values in the FloatArray within the range [min, max].

cosineSimilarity

fun FloatArray.cosineSimilarity(other: FloatArray): Float

Calculates the Cosine Similarity between two FloatArray vectors.

dequantize

fun ByteArray.dequantize(scale: Float, zeroPoint: Int): FloatArray

Dequantizes the ByteArray into a FloatArray using the given scale and zeroPoint.

euclideanDistance

fun FloatArray.euclideanDistance(other: FloatArray): Float

Calculates the Euclidean Distance between two FloatArray vectors.

getInputBuffer

suspend fun LiteRTCompiler.getInputBuffer(index: Int): TFBuffer

Returns the TFBuffer for the input tensor at the specified index.

getOutputBuffer

suspend fun LiteRTCompiler.getOutputBuffer(index: Int): TFBuffer

Returns the TFBuffer for the output tensor at the specified index.

mean

fun FloatArray.mean(): Float

Calculates the mean value of the FloatArray.

normalize

fun FloatArray.normalize(mean: Float, std: Float): FloatArray

Normalizes the FloatArray using the given mean and std.

pad

fun FloatArray.pad(before: Int, after: Int, value: Float = 0.0f): FloatArray

Adds padding to the FloatArray before and after its current content.

performNms

fun performNms(boxes: Array<FloatArray>, scores: FloatArray, iouThreshold: Float): List<Int>

Performs Non-Maximum Suppression (NMS) on a set of bounding boxes.

quantize

fun FloatArray.quantize(scale: Float, zeroPoint: Int): ByteArray

Quantizes the FloatArray into a ByteArray using the given scale and zeroPoint.

readFloatValue

suspend fun TFBuffer.readFloatValue(): Float

Reads a single Float value from the TFBuffer.

reshape

fun FloatArray.reshape(vararg dims: Int): FloatArray

Simulates a reshape operation on the FloatArray.

sigmoid

fun FloatArray.sigmoid(): FloatArray

Computes the Sigmoid function for each element in the FloatArray.

softmax

fun FloatArray.softmax(): FloatArray

Computes the Softmax probabilities for the FloatArray.

std

fun FloatArray.std(): Float

Calculates the standard deviation of the FloatArray.

toBooleanArray

suspend fun TFBuffer.toBooleanArray(): BooleanArray

Reads the content of the TFBuffer as a BooleanArray.

fun ByteArray.toBooleanArray(): BooleanArray

Converts a ByteArray to a BooleanArray (true if non-zero).

toByteArray

suspend fun TFBuffer.toByteArray(): ByteArray

Reads the content of the TFBuffer as a ByteArray.

fun BooleanArray.toByteArray(): ByteArray

Converts a BooleanArray to a ByteArray (1 for true, 0 for false).

fun FloatArray.toByteArray(): ByteArray

Converts a FloatArray to a ByteArray by rounding each element to the nearest Byte.

toCategories

fun FloatArray.toCategories(labels: List<String>, threshold: Float = 0.0f): List<LiteRTExt.Category>

Converts the FloatArray scores into a list of LiteRTExt.Category.

toFloatArray

suspend fun TFBuffer.toFloatArray(): FloatArray

Reads the content of the TFBuffer as a FloatArray.

fun ByteArray.toFloatArray(): FloatArray

Converts a ByteArray to a FloatArray.

fun IntArray.toFloatArray(): FloatArray

Converts an IntArray to a FloatArray.

toIntArray

suspend fun TFBuffer.toIntArray(): IntArray

Reads the content of the TFBuffer as an IntArray.

fun FloatArray.toIntArray(): IntArray

Converts a FloatArray to an IntArray by rounding each element to the nearest Int.

toLongArray

suspend fun TFBuffer.toLongArray(): LongArray

Reads the content of the TFBuffer as a LongArray.

topK

fun FloatArray.topK(k: Int): List<Pair<Int, Float>>

Returns the top k elements as a list of Pair(index, score).

use

suspend fun <T> LiteRTCompiler.use(block: suspend (LiteRTCompiler) -> T): T

Automatically closes the LiteRTCompiler after the block is executed.

writeFloatValue

fun TFBuffer.writeFloatValue(value: Float)

Writes a single Float value into the TFBuffer.

writeTo

fun BooleanArray.writeTo(buffer: TFBuffer)

Writes the BooleanArray data into the TFBuffer.

fun ByteArray.writeTo(buffer: TFBuffer)

Writes the ByteArray data into the TFBuffer.

fun FloatArray.writeTo(buffer: TFBuffer)

Writes the FloatArray data into the TFBuffer.

fun IntArray.writeTo(buffer: TFBuffer)

Writes the IntArray data into the TFBuffer.

fun LongArray.writeTo(buffer: TFBuffer)

Writes the LongArray data into the TFBuffer.