summaryrefslogtreecommitdiff
path: root/2005/i/robert/src/ovision/see/n2ml.cc
diff options
context:
space:
mode:
Diffstat (limited to '2005/i/robert/src/ovision/see/n2ml.cc')
-rw-r--r--2005/i/robert/src/ovision/see/n2ml.cc44
1 files changed, 44 insertions, 0 deletions
diff --git a/2005/i/robert/src/ovision/see/n2ml.cc b/2005/i/robert/src/ovision/see/n2ml.cc
new file mode 100644
index 0000000..730d05e
--- /dev/null
+++ b/2005/i/robert/src/ovision/see/n2ml.cc
@@ -0,0 +1,44 @@
+// n2ml.cc
+// robert - Programme du robot APBteam
+// Copyright (C) 2005 Olivier Gaillard
+#include <iostream>
+#include <vector>
+#include <string>
+
+#include "imgInterface.hh"
+#include "segmLearn.hh"
+#include "oconfig.hh"
+
+
+int main(int argc, char **argv)
+{
+ // Usage
+ if (argc < 4)
+ {
+ std::cout << "Usage : n2ml <output number> {rgb| yuv} <file list>" << std::endl;
+ return 1;
+ }
+ int nbOutput = atoi(argv[1]);
+ // Parse de la ligne de commande
+ std::string *str = new std::string (argv[2]);
+ Image::PixelFormat pf;
+ if (*str == "rgb") pf = Image::rgb;
+ else if (*str == "yuv") pf = Image::yuv;
+ else std::cerr << "Wrong pixelformat\n";
+ std::vector<std::string> list;
+ while (argc-- > 3)
+ {
+ delete str;
+ str = new std::string (argv[argc]);
+ list.push_back(*str);
+ }
+ // Initialisation des classes
+ OConfig oconfig;
+ Img img;
+ img.loadMultipeRaw (list, pf, 360, 296);
+ SegmLearn segm;
+ segm.buildNN (nbOutput, Segm::generate);
+ segm.trainNN (&img);
+ segm.save ();
+}
+