Package rawDeepLearningClassifier.dlClassification.ketos

Class KetosClassifier2

java.lang.Object

rawDeepLearningClassifier.dlClassification.StandardClassifierModel

rawDeepLearningClassifier.dlClassification.archiveModel.ArchiveModelClassifier

rawDeepLearningClassifier.dlClassification.ketos.KetosClassifier2

All Implemented Interfaces:

PamSettings, SettingsNameProvider, DLClassiferModel

public class KetosClassifier2
extends ArchiveModelClassifier

Classifier which uses deep learning models from Meridian's Ketos framework.

Ketos uses TensorFlow models and packages them inside a zipped .ktpb file which contains a JSON file for the transforms and a .pb model. Users can select a .ktpb file - PAMGaurd will decompress it, find the JSON file, set up the transforms and load the model.

Details on Meridians framework can be found at https://meridian.cs.dal.ca/2015/04/12/ketos/

KetosClassifier2 is a more abstracted version of KetosClassifer which inherits most functionality from ArchiveModel

Author:

Jamie Macaulay

Nested Class Summary

Nested classes/interfaces inherited from class rawDeepLearningClassifier.dlClassification.StandardClassifierModel

StandardClassifierModel.TaskThread

Field Summary

Fields

Modifier and Type	Field	Description
static String	MODEL_NAME

Constructor Summary

Constructors

Constructor	Description
KetosClassifier2(DLControl dlControl)

Method Summary

Modifier and Type	Method	Description
String[]	getFileExtensions()	Get the file extensions for the model type.
ArchiveModelWorker	getModelWorker()	Get the KetosWorker.
String	getName()	Get the name of the model.
StandardModelParams	makeParams()	Create the parameters class for the model.

Methods inherited from class rawDeepLearningClassifier.dlClassification.archiveModel.ArchiveModelClassifier

getDLModelSettings, getDLParams, getDLWorker, getKetosParams, getModelUI, getSettingsReference, getSettingsVersion, getUnitName, getUnitType, isModelType, restoreSettings, setDLParams

Methods inherited from class rawDeepLearningClassifier.dlClassification.StandardClassifierModel

checkSettingsOK, closeModel, getClassNameIDs, getClassNames, getDLControl, getModelStatus, getNumClasses, getRawSettingsPane, isDecision, isModelExtensions, millis2Samples, prepModel, runModel, setModel

Methods inherited from class java.lang.Object

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

Field Details

MODEL_NAME

public static String MODEL_NAME

Constructor Details

KetosClassifier2

public KetosClassifier2(DLControl dlControl)

Method Details

getFileExtensions

public String[] getFileExtensions()

Description copied from class: ArchiveModelClassifier

Get the file extensions for the model type.

Overrides:

getFileExtensions in class ArchiveModelClassifier

Returns:

the file extension.

getName

public String getName()

Description copied from interface: DLClassiferModel

Get the name of the model.

Specified by:

getName in interface DLClassiferModel

Overrides:

getName in class ArchiveModelClassifier

Returns:

the name of the model.

getModelWorker

public ArchiveModelWorker getModelWorker()

Description copied from class: ArchiveModelClassifier

Get the KetosWorker. this handles loading and running the Ketos model.

Overrides:

getModelWorker in class ArchiveModelClassifier

Returns:

the Ketos worker.

makeParams

public StandardModelParams makeParams()

Create the parameters class for the model. This can be overridden for bespoke parameters. classes.

Overrides:

makeParams in class ArchiveModelClassifier

Returns:

a new parameters class object.