deepwater.backends.mxnet

Class MXNetBackend

java.lang.Object

deepwater.backends.mxnet.MXNetBackend

All Implemented Interfaces:

deepwater.backends.BackendTrain

public class MXNetBackend
extends java.lang.Object
implements deepwater.backends.BackendTrain

Constructor Summary

Constructors

Constructor and Description
MXNetBackend()

Method Summary

Modifier and Type	Method and Description
deepwater.backends.BackendModel	buildNet(deepwater.datasets.ImageDataSet dataset, deepwater.backends.RuntimeOptions opts, deepwater.backends.BackendParams bparms, int num_classes, java.lang.String name)
void	delete(deepwater.backends.BackendModel m)
float[]	loadMeanImage(deepwater.backends.BackendModel m, java.lang.String param_path)
void	loadParam(deepwater.backends.BackendModel m, java.lang.String networkParms)
float[]	predict(deepwater.backends.BackendModel m, float[] data)
float[]	predict(deepwater.backends.BackendModel m, float[] data, float[] label)
void	saveModel(deepwater.backends.BackendModel m, java.lang.String model_path)
void	saveParam(deepwater.backends.BackendModel m, java.lang.String param_path)
void	setParameter(deepwater.backends.BackendModel m, java.lang.String name, float value)
java.lang.String	toJson(deepwater.backends.BackendModel m)
float[]	train(deepwater.backends.BackendModel m, float[] data, float[] label)

Methods inherited from class java.lang.Object

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

Constructor Detail

MXNetBackend

public MXNetBackend()

Method Detail

delete

public void delete(deepwater.backends.BackendModel m)

Specified by:

delete in interface deepwater.backends.BackendTrain

buildNet

public deepwater.backends.BackendModel buildNet(deepwater.datasets.ImageDataSet dataset,
                                                deepwater.backends.RuntimeOptions opts,
                                                deepwater.backends.BackendParams bparms,
                                                int num_classes,
                                                java.lang.String name)

Specified by:

buildNet in interface deepwater.backends.BackendTrain

setParameter

public void setParameter(deepwater.backends.BackendModel m,
                         java.lang.String name,
                         float value)

Specified by:

setParameter in interface deepwater.backends.BackendTrain

train

public float[] train(deepwater.backends.BackendModel m,
                     float[] data,
                     float[] label)

Specified by:

train in interface deepwater.backends.BackendTrain

predict

public float[] predict(deepwater.backends.BackendModel m,
                       float[] data,
                       float[] label)

Specified by:

predict in interface deepwater.backends.BackendTrain

predict

public float[] predict(deepwater.backends.BackendModel m,
                       float[] data)

Specified by:

predict in interface deepwater.backends.BackendTrain

loadParam

public void loadParam(deepwater.backends.BackendModel m,
                      java.lang.String networkParms)

Specified by:

loadParam in interface deepwater.backends.BackendTrain

saveModel

public void saveModel(deepwater.backends.BackendModel m,
                      java.lang.String model_path)

Specified by:

saveModel in interface deepwater.backends.BackendTrain

saveParam

public void saveParam(deepwater.backends.BackendModel m,
                      java.lang.String param_path)

Specified by:

saveParam in interface deepwater.backends.BackendTrain

loadMeanImage

public float[] loadMeanImage(deepwater.backends.BackendModel m,
                             java.lang.String param_path)

Specified by:

loadMeanImage in interface deepwater.backends.BackendTrain

toJson

public java.lang.String toJson(deepwater.backends.BackendModel m)

Specified by:

toJson in interface deepwater.backends.BackendTrain