C:/Users/albertlr/projects/TarzaNN/TarzaNN/InputLearningFeaturePlane.h

Go to the documentation of this file.

#ifndef INPUTLEARNINGFEATUREPLANE_H_INCLUDED
#define INPUTLEARNINGFEATUREPLANE_H_INCLUDED

#include  <qimage.h>

#include "nowarn.h"
#include "common.h"
#include <stdlib.h>

#include "OutputImage.h"
#include "InputFeaturePlaneAbstract.h"
#include "FeaturePlaneInput.h"
#include "NotifyStrategyASync.h"
#include "NotifyStrategySync.h"

#include "FacesDataset.h"
#include "FullNorbDataset.h"
#include "GreebleDataset.h"
#include "RoozDataset.h"
#include "RoozDataset2.h"
#include "RoozDatasetOR2.h"
#include "RoozDataset5.h"
#include "SmallNorbDataset.h"
#include "SOMDataset.h"
#include "SOMGeomDataset.h"
#include "VanRullenDataset.h"

//#define SIMPLE_TEST

class InputLearningFeaturePlane : public InputFeaturePlaneAbstract {
        friend class FeaturePlaneFactory;
public:
    ~InputLearningFeaturePlane();
        OutputImage* getCurrentOutput();
        OutputImage* getCurrentOutput(int index);
        OutputImage** getCurrentOutputs();
        void inhibitReturn();
    
    bool isSameAsLast();


protected:
        char* mask_filename;
        bool input_scale;
        float input_min, input_max;
        int window;
        //should noise be added to the input image after loading
        bool noise;
        
        InputLearningFeaturePlane( Network* net, int l, char* filename, QString* name, bool isWTA, bool visible, bool scale, bool noise, float min, float max, NotifyStrategyAbstract* notify, int maxSteps, int levels);

private:

        void step();
        void applyGatingControl();
        Dataset* dataset;
        int currentIndex;
        void loadDataset(char* inputFileName);
#ifdef SIMPLE_TEST
        //test
        QImage* trainTestImages[5];
#endif
};
#endif //INPUTLEARNINGFEATUREPLANE_H_INCLUDED