fast::ImageClassificationNetwork class

Image classification neural network.

This class is a convenience class for a neural network which performs image classification. Use setLabels method to define the class names. The output is then ImageClassification which is a map from class names to confidence values.

Base classes

class NeuralNetwork
Neural network process object.

Public functions

auto create(std::string modelFilename, std::vector<std::string> labels, float scaleFactor, float meanIntensity, float stanardDeviationIntensity, int temporalWindow, std::vector<NeuralNetworkNode> inputNodes, std::vector<NeuralNetworkNode> outputNodes, std::string inferenceEngine, std::vector<std::string> customPlugins) FAST_CONSTRUCTOR(ImageClassificationNetwork -> std::shared_ptr<ImageClassificationNetwork>
void setLabels(std::vector<std::string> labels)
void loadAttributes() virtual

Public variables

modelFilename
inputNodes
outputNodes
inferenceEngine
customPlugins

Private functions

void execute() virtual