// robert - programme du robot 2005 // // Copyright (C) 2005 Olivier Gaillard // // Robot APB Team/Efrei 2005. // Web: http://assos.efrei.fr/robot/ // Email: robot AT efrei DOT fr // // This program is free software; you can redistribute it and/or modify // it under the terms of the GNU General Public License as published by // the Free Software Foundation; either version 2 of the License, or // (at your option) any later version. // // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU General Public License for more details. // // You should have received a copy of the GNU General Public License // along with this program; if not, write to the Free Software // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. #ifndef segmLearn_h #define segmLearn_h #include "segm.hh" /// Segmente l'image avec le réseau de neurones class SegmLearn : public Segm { /// memorise la popularité des noeuds pour enlever les sorties inefficaces int *freq_; public: /// Constructeur SegmLearn (); /// Destructeur virtual ~SegmLearn (void); /// Apprentissage du NN void trainNN (Img *img); /// Cree le NN void buildNN (const int nbOutput, const bool loadFromFile = false); /// Renvoie le code la couleur segmentee virtual unsigned char giveColor (const unsigned char *x, const bool testUndefined = false, const bool indexProcess = false); private: /// Genere des poids pour un noeud void weightsGen (const int numNode); /// Renvoie le code de la couleur segmentee unsigned char findColorNN (const unsigned char *x, const bool testOutputMax = false) const; }; #endif // segmLearn_h