Package-level declarations

Types

Link copied to clipboard
abstract class BaseModelParser : ModelParser

Abstract base class for model parsers that provides common functionality for validation, error handling, and tensor size calculations.

Link copied to clipboard
object ClassNames

Common class name sets for object detection and classification models.

Link copied to clipboard

Unified data type representation across all model formats (ONNX, GGUF, SafeTensors, etc.).

Link copied to clipboard

Bidirectional mapping utility between SKaiNET DType and unified DataType.

Link copied to clipboard
sealed class LoadingProgress

Represents progress within a loading operation.

Link copied to clipboard
data class LoadingStage(val id: String, val name: String, val order: Int, val totalStages: Int)

Represents a stage in the loading process.

Link copied to clipboard

Supported model file formats.

Link copied to clipboard
data class ModelInfo(val format: ModelFormat, val version: String? = null, val producer: String? = null, val domain: String? = null, val irVersion: Long? = null, val additionalMetadata: Map<String, Any> = emptyMap())

General model information extracted from model files.

Link copied to clipboard
data class ModelMetadata(val format: ModelFormat, val isValid: Boolean, val errorMessage: String? = null)

Result of parsing model metadata.

Link copied to clipboard
interface ModelParser

Common interface for all model format parsers.

Link copied to clipboard

Base class for model parsing errors. Provides user-friendly error messages for different failure scenarios.

Link copied to clipboard

Interface for loaders that support progress reporting.

Link copied to clipboard
data class TensorInfo(val name: String, val shape: List<Long>, val dataType: DataType, val elementCount: Long, val sizeInBytes: Long?, val format: ModelFormat, val nativeDType: String? = null, val skainetDType: String? = null, val canLoadNatively: Boolean = false, val sourceFile: String? = null, val shardIndex: Int? = null, val totalShards: Int? = null)

Unified tensor information across all model formats.