hcm.ssj.ml

Class TensorFlow

java.lang.Object

hcm.ssj.core.Component

hcm.ssj.ml.Model

hcm.ssj.ml.TensorFlow

All Implemented Interfaces:

java.lang.Runnable

public class TensorFlow
extends Model

TensorFlow model. Supports prediction using tensor flow's Android API. Requires pre-trained frozen graph, e.g. using SSI.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	TensorFlow.Options

Field Summary

Fields

Modifier and Type	Field and Description
TensorFlow.Options	options

Fields inherited from class hcm.ssj.ml.Model

class_names, dirPath, input_bytes, input_dim, input_sr, input_type, isTrained, modelFileName, modelOptionFileName, n_classes, select_dimensions

Fields inherited from class hcm.ssj.core.Component

_evchannel_in, _evchannel_out, _isSetup, _name, _safeToKill, _terminate, threadPriority

Constructor Summary

Constructors

Constructor and Description
TensorFlow()

Method Summary

Modifier and Type	Method and Description
protected float[]	forward(Stream stream) forward data to the model for classification/inference
Model.Options	getOptions()
protected void	loadModel(java.io.File file) Load model from file
protected void	loadOption(java.io.File file) Load model options from file

Methods inherited from class hcm.ssj.ml.Model

getClassNames, getInputDim, getInputSr, getInputType, getName, getNumClasses, isTrained, load, run, save, setClassNames, setNumClasses, setup, setup, train, validateInput, waitUntilReady

Methods inherited from class hcm.ssj.core.Component

clear, close, forcekill, getComponentName, getEventChannelOut, isSetup, reset

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

options

public TensorFlow.Options options

Constructor Detail

TensorFlow

public TensorFlow()

Method Detail

getOptions

public Model.Options getOptions()

Specified by:

getOptions in class Model

forward

protected float[] forward(Stream stream)

Description copied from class: Model

forward data to the model for classification/inference

Parameters:

stream - Stream

Returns:

double[] classification/inference probabilities as outputed by the model

loadOption

protected void loadOption(java.io.File file)

Description copied from class: Model

Load model options from file

loadModel

protected void loadModel(java.io.File file)

Description copied from class: Model

Load model from file