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.
- Alphabetic
- By Inheritance
- ONNXModel
- Serializable
- Serializable
- ComplexParamsReadable
- MLReadable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
def
load(path: String): ONNXModel
- Definition Classes
- MLReadable
- Annotations
- @Since( "1.6.0" )
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
def
read: MLReader[ONNXModel]
- Definition Classes
- ComplexParamsReadable → MLReadable
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
def
toString(): String
- Definition Classes
- AnyRef → Any
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()