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

Go to the documentation of this file.

/****************************************************************************
 **
 ** Copyright C 2002-2012 Laboratory for Active and Attentive Vision (LAAV), Department of Computer Science and Engineering, York University, Toronto, ON, Canada.
 ** All rights reserved.
 **
 ** This file is part of the TarzaNN Neural Network Simulator.
 **
 ** This file may be distributed and/or modified under the terms of the
 ** GNU General Public License version 2 as published by the Free Software
 ** Foundation and appearing in the file LICENSE.GPL included in the
 ** packaging of this file.
 **
 ** See http://www.tarzann.org/gpl/ for GPL licensing information.
 **
 ** Contact info@tarzann.org if any conditions of this licensing are
 ** not clear to you.
 **
 ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 **
 ****************************************************************************/

#ifndef SOMDataset_H_INCLUDED
#define SOMDataset_H_INCLUDED

#include <vector>
using namespace std;
#include "Dataset.h"
#include <QDir>

#define STEPS_STAGE_1 10000
#define STEPS_STAGE_2 10000
#define STEPS_STAGE_3 10000

#define IMAGE_SIZE_X 54
#define IMAGE_SIZE_Y 54

//Dataset for SOM learning - based on Risto Miikkulainen, James A. Bednar, Yoonsuck Choe, and Joseph Sirosh. Computational Maps in the Visual Cortex. Springer, Berlin, 2005.
class SOM_Dataset : public Dataset {
        
private:
        long imageBaseOffset;
        long imageSize;
        FILE* imageFile;
        char* currentfileName;
        QFileInfoList fileList;
        Matrix* result;
        
public:
        SOM_Dataset(char* basefilename);
        ~SOM_Dataset();
        OutputImage* getImage(int index, OutputImage** list, int LorR=0);
        Matrix* getDesiredOutcome(int index);
        int getRandomIndex();
};
#endif/*SOMDataset_H_INCLUDED*/