skainet-io-onnx/sk.ainet.io.onnx/OnnxModelMetadata

OnnxModelMetadata

data class OnnxModelMetadata(val irVersion: Long, val producerName: String?, val producerVersion: String?, val modelVersion: Long, val domain: String?, val docString: String?, val classNames: Map<Int, String>?, val numClasses: Int?, val description: String?, val author: String?, val license: String?, val task: String?, val rawMetadata: Map<String, String>)(source)

Parsed model metadata from an ONNX file.

This class extracts common metadata fields from ONNX ModelProto's metadataProps in a structured, type-safe manner. Supports various model types including object detectors (YOLO), classifiers, etc.

Usage:

val modelBytes = File("model.onnx").readBytes()
val modelProto = ModelProto.decodeFromByteArray(modelBytes)
val metadata = OnnxModelMetadata.from(modelProto)

println("Classes: ${metadata.classNames?.size ?: "none"}")
metadata.classNames?.forEach { (id, name) -> println("$id: $name") }

Constructors

OnnxModelMetadata

constructor(irVersion: Long, producerName: String?, producerVersion: String?, modelVersion: Long, domain: String?, docString: String?, classNames: Map<Int, String>?, numClasses: Int?, description: String?, author: String?, license: String?, task: String?, rawMetadata: Map<String, String>)