org.opencv.dnn

Class ClassificationModel

java.lang.Object

org.opencv.dnn.Net

org.opencv.dnn.Model

org.opencv.dnn.ClassificationModel

public class ClassificationModel
extends Model

This class represents high-level API for classification models. ClassificationModel allows to set params for preprocessing input image. ClassificationModel creates net from file with trained weights and config, sets preprocessing input, runs forward pass and return top-1 prediction.

Constructor Summary

Constructors

Constructor and Description
ClassificationModel(Net network) Create model from deep learning network.
ClassificationModel(java.lang.String model) Create classification model from network represented in one of the supported formats.
ClassificationModel(java.lang.String model, java.lang.String config) Create classification model from network represented in one of the supported formats.

Method Summary

Modifier and Type	Method and Description
static ClassificationModel	__fromPtr__(long addr)
void	classify(Mat frame, int[] classId, float[] conf)

Methods inherited from class org.opencv.dnn.Model

predict, setInputCrop, setInputMean, setInputParams, setInputParams, setInputParams, setInputParams, setInputParams, setInputParams, setInputScale, setInputSize, setInputSize, setInputSwapRB

Methods inherited from class org.opencv.dnn.Net

connect, dump, dumpToFile, empty, enableFusion, forward, forward, forward, forward, forward, getFLOPS, getFLOPS, getFLOPS, getFLOPS, getLayer, getLayerId, getLayerNames, getLayersCount, getLayerTypes, getMemoryConsumption, getMemoryConsumption, getMemoryConsumption, getNativeObjAddr, getParam, getParam, getPerfProfile, getUnconnectedOutLayers, getUnconnectedOutLayersNames, readFromModelOptimizer, readFromModelOptimizer, setHalideScheduler, setInput, setInput, setInput, setInput, setInputsNames, setParam, setPreferableBackend, setPreferableTarget

Methods inherited from class java.lang.Object

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

Constructor Detail

ClassificationModel

public ClassificationModel(Net network)

Create model from deep learning network.

Parameters:

network - Net object.

ClassificationModel

public ClassificationModel(java.lang.String model)

Create classification model from network represented in one of the supported formats. An order of model and config arguments does not matter.

Parameters:

model - Binary file contains trained weights.

ClassificationModel

public ClassificationModel(java.lang.String model,
                           java.lang.String config)

Create classification model from network represented in one of the supported formats. An order of model and config arguments does not matter.

Parameters:

model - Binary file contains trained weights.

config - Text file contains network configuration.

Method Detail

__fromPtr__

public static ClassificationModel __fromPtr__(long addr)

classify

public void classify(Mat frame,
                     int[] classId,
                     float[] conf)