iRoCS Toolbox  1.1.0
TrainingParameters.hh
Go to the documentation of this file.
1 /**************************************************************************
2  *
3  * Copyright (C) 2015 Thorsten Falk
4  *
5  * Image Analysis Lab, University of Freiburg, Germany
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  *
21  **************************************************************************/
22 
23 #ifndef TRAININGPARAMETERS_HH
24 #define TRAININGPARAMETERS_HH
25 
26 #ifdef HAVE_CONFIG_H
27 #include <config.hh>
28 #endif
29 
30 #include <string>
31 #include <vector>
32 
33 #include "iRoCSFeatures.hh"
34 
36 
38 {
39 
40 public:
41 
43  virtual ~TrainingParameters();
44 
45  void addTrainFile(TrainfileParameters* parameters);
46  virtual std::vector<TrainfileParameters*> trainFiles() const;
47 
48  virtual void setFeatureGroup(std::string const &group);
49  virtual std::string featureGroup() const;
50 
51  virtual void setGenerateRandomSamples(bool generate);
52  virtual bool generateRandomSamples() const;
53 
54  virtual void setNInRootSamples(int samples);
55  virtual int nInRootSamples() const;
56 
57  virtual void setNOutRootSamples(int samples);
58  virtual int nOutRootSamples() const;
59 
60  virtual void setModelFileName(std::string const &name);
61  virtual std::string modelFileName() const;
62 
63  virtual void setSdFeatureNormalization(
66 
67  virtual void setHoughFeatureNormalization(
70 
71  virtual void setCost(double cost);
72  virtual double cost() const;
73 
74  virtual void setGamma(double gamma);
75  virtual double gamma() const;
76 
77  std::string check();
78 
79 private:
80 
81  std::vector<TrainfileParameters*> _trainFiles;
82  std::string _featureGroup;
83  bool _generateRandomSamples;
84  int _nInRootSamples, _nOutRootSamples;
85  std::string _modelFileName;
86  iRoCS::Features::NormalizationType _sdNormalization, _houghNormalization;
87  double _cost, _gamma;
88 
89 };
90 
91 #endif
virtual bool generateRandomSamples() const
virtual void setGamma(double gamma)
virtual int nInRootSamples() const
virtual void setHoughFeatureNormalization(iRoCS::Features::NormalizationType normType)
virtual void setNInRootSamples(int samples)
virtual void setCost(double cost)
virtual void setSdFeatureNormalization(iRoCS::Features::NormalizationType normType)
virtual ~TrainingParameters()
std::string check()
virtual void setFeatureGroup(std::string const &group)
virtual double gamma() const
virtual void setModelFileName(std::string const &name)
void addTrainFile(TrainfileParameters *parameters)
virtual std::string modelFileName() const
virtual int nOutRootSamples() const
virtual double cost() const
virtual iRoCS::Features::NormalizationType houghFeatureNormalization() const
virtual std::string featureGroup() const
virtual void setNOutRootSamples(int samples)
virtual iRoCS::Features::NormalizationType sdFeatureNormalization() const
virtual std::vector< TrainfileParameters * > trainFiles() const
virtual void setGenerateRandomSamples(bool generate)