00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkLevenbergMarquardtOptimizer_h
00018 #define __itkLevenbergMarquardtOptimizer_h
00019
00020 #include "itkMultipleValuedNonLinearVnlOptimizer.h"
00021 #include "vnl/algo/vnl_levenberg_marquardt.h"
00022
00023 namespace itk
00024 {
00025
00031 class ITK_EXPORT LevenbergMarquardtOptimizer :
00032 public MultipleValuedNonLinearVnlOptimizer
00033 {
00034 public:
00036 typedef LevenbergMarquardtOptimizer Self;
00037 typedef MultipleValuedNonLinearVnlOptimizer Superclass;
00038 typedef SmartPointer<Self> Pointer;
00039 typedef SmartPointer<const Self> ConstPointer;
00040
00042 itkNewMacro(Self);
00043
00045 itkTypeMacro( LevenbergMarquardtOptimizer, MultipleValuedNonLinearVnlOptimizer );
00046
00048 typedef vnl_vector<double> InternalParametersType;
00049
00051 typedef vnl_levenberg_marquardt InternalOptimizerType;
00052
00054 vnl_levenberg_marquardt * GetOptimizer(void);
00055
00057 void StartOptimization( void );
00058
00060 virtual void SetCostFunction( MultipleValuedCostFunction * costFunction );
00061
00062 void SetNumberOfIterations(unsigned int iterations);
00063 void SetValueTolerance(double tol);
00064 void SetGradientTolerance(double tol);
00065 void SetEpsilonFunction(double epsilon);
00066
00068 MeasureType GetValue();
00069
00070 protected:
00071 LevenbergMarquardtOptimizer();
00072 virtual ~LevenbergMarquardtOptimizer();
00073
00074 typedef Superclass::CostFunctionAdaptorType CostFunctionAdaptorType;
00075
00076 private:
00077 LevenbergMarquardtOptimizer(const Self&);
00078 void operator=(const Self&);
00079
00080 bool m_OptimizerInitialized;
00081 InternalOptimizerType * m_VnlOptimizer;
00082 unsigned int m_NumberOfIterations;
00083 double m_ValueTolerance;
00084 double m_GradientTolerance;
00085 double m_EpsilonFunction;
00086
00087 };
00088
00089 }
00090
00091
00092 #endif
00093
00094
00095
00096