summaryrefslogtreecommitdiff
path: root/2005/i/robert/src/ovision/see/segmLearn.hh
diff options
context:
space:
mode:
Diffstat (limited to '2005/i/robert/src/ovision/see/segmLearn.hh')
-rw-r--r--2005/i/robert/src/ovision/see/segmLearn.hh53
1 files changed, 53 insertions, 0 deletions
diff --git a/2005/i/robert/src/ovision/see/segmLearn.hh b/2005/i/robert/src/ovision/see/segmLearn.hh
new file mode 100644
index 0000000..3d0f564
--- /dev/null
+++ b/2005/i/robert/src/ovision/see/segmLearn.hh
@@ -0,0 +1,53 @@
+// 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