ImageClassificationNetwork.hpp

Go to the documentation of this file.

#pragma once
 
#include "FAST/ProcessObject.hpp"
#include "NeuralNetwork.hpp"
#include "FAST/Data/SimpleDataObject.hpp"
#include <queue>
 
namespace fast {
 
// Create the data object used as output from the ImageClassificationNetwork
 
typedef std::map<std::string, float> classifications;
FAST_SIMPLE_DATA_OBJECT(ImageClassification, classifications)
 
 
class FAST_EXPORT ImageClassificationNetwork : public NeuralNetwork {
    FAST_OBJECT(ImageClassificationNetwork)
    public:
        void setLabels(std::vector<std::string> labels);
        void loadAttributes();
    private:
        ImageClassificationNetwork();
        void execute();
 
        // A map of label -> score
        std::vector<std::string> mLabels;
 
};
 
 
class FAST_EXPORT  ClassificationToText : public ProcessObject {
    FAST_OBJECT(ClassificationToText)
    private:
        std::deque<std::map<std::string, float>> mBuffer; // used for calculating temporal average
        int mBufferSize = 1; // How large the buffer can be
 
        ClassificationToText();
        void loadAttributes();
        void execute();
};
 
}