Packages

object ONNXModel extends ComplexParamsReadable[ONNXModel] with Serializable

Object model for an ONNX model: OrtSession

|-InputInfo: Map[String, NodeInfo]

|-OutputInfo: Map[String, NodeInfo] OrtSession is the entry point for the object model. Most importantly it defines the InputInfo and OutputInfo maps. ------------------------------------ NodeInfo

|-name: String

|-info: ValueInfo Each NodeInfo is a name and ValueInfo tuple. ValueInfo has three implementations, explained below. ------------------------------------ TensorInfo extends ValueInfo

|-shape: Array[Long]

|-type: OnnxJavaType TensorInfo is the most common type of ValueInfo. It defines the type of the tensor elements, and the shape. The first dimension of the tensor is assumed to be the batch size. For example, FLOAT[-1, 3, 224, 224] could represent a unlimited batch size * 3 channels * 224 height * 224 width tensor, where each element is a float. ------------------------------------ SequenceInfo extends ValueInfo

|-sequenceOfMaps: Boolean

|-sequenceType: OnnxJavaType

|-mapInfo: MapInfo

|-length: Int SequenceInfo can be a sequence of values (value type specified by sequenceType) if sequenceOfMaps is false, or a sequence of MapInfo if sequenceOfMaps is true. Sequence of MapInfo is usually used for ZipMap type of output, where the sequence represent the batch, and each MapInfo represents probability or logits outcome per class for each observation. ------------------------------------ MapInfo extends ValueInfo

|-keyType: OnnxJavaType

|-valueType: OnnxJavaType

|-size: Int MapInfo defines keyType, valueType and size. It is usually used inside SequenceInfo.

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ONNXModel
  2. Serializable
  3. Serializable
  4. ComplexParamsReadable
  5. MLReadable
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  6. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  10. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  11. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  12. def load(path: String): ONNXModel
    Definition Classes
    MLReadable
    Annotations
    @Since( "1.6.0" )
  13. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  14. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  15. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  16. def read: MLReader[ONNXModel]
    Definition Classes
    ComplexParamsReadable → MLReadable
  17. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  18. def toString(): String
    Definition Classes
    AnyRef → Any
  19. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  20. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  21. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from Serializable

Inherited from Serializable

Inherited from MLReadable[ONNXModel]

Inherited from AnyRef

Inherited from Any

Ungrouped