object OpenvinoWrapper extends Serializable
Companion object
- Alphabetic
- By Inheritance
- OpenvinoWrapper
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Type Members
- case class DecoderWrappers(decoder: OpenvinoWrapper) extends Product with Serializable
- case class EncoderDecoderWithoutPastWrappers(encoder: OpenvinoWrapper, decoder: OpenvinoWrapper) extends Product with Serializable
- case class EncoderDecoderWrappers(encoder: OpenvinoWrapper, decoder: OpenvinoWrapper, decoderWithPast: OpenvinoWrapper) extends Product with Serializable
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()
-
def
convertToOpenvinoFormat(srcPath: String, exportPath: String, compressToFp16: Boolean = false): (Path, Path)
Convert the model at srcPath to OpenVINO IR Format and export to exportPath.
Convert the model at srcPath to OpenVINO IR Format and export to exportPath.
- srcPath
Path to the source model
- exportPath
Path to export converted model to
- compressToFp16
Whether to perform weight compression to FP16
- returns
Paths to the exported XML and BIN files
-
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
-
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(sparkSession: SparkSession, modelPath: String, modelName: String = Openvino.ovModel, zipped: Boolean = true, useBundle: Boolean = false, detectedEngine: String = Openvino.name, properties: Map[String, String] = Map.empty, ovFileSuffix: Option[String] = None): OpenvinoWrapper
Read the model from the given path, unpack if zipped, and return the loaded OpenvinoWrapper.
Read the model from the given path, unpack if zipped, and return the loaded OpenvinoWrapper. If source model is not in OpenVINO format, it is converted first.
- sparkSession
The Spark Session
- modelPath
Path to the model
- modelName
The model filename
- zipped
Unpack zipped model
- useBundle
Load exported model
- detectedEngine
The source model format
- properties
Properties for this load operation
- returns
The resulting OpenVINO model wrapper
-
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()
-
def
withSafeOvModelLoader(modelPath: Option[String] = None, device: String = "CPU", properties: Map[String, String]): CompiledModel
Prepare the model for inference by compiling into a device-specific graph representation.
Prepare the model for inference by compiling into a device-specific graph representation. Returns the compiled model object.
- modelPath
Optional path to the model directory
- device
Device to compile the model to
- properties
Properties for this load operation
- returns
Object representing the compiled model