itkweightsetbase.h

DTMK软件开发包,此为开源软件,是一款很好的医学图像开发资源.

/*=========================================================================

Program:   Insight Segmentation & Registration Toolkit
Module:    $RCSfile: itkWeightSetBase.h,v $
Language:  C++
Date:      $Date: 2007-08-17 13:10:57 $
Version:   $Revision: 1.9 $

Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#ifndef __itkWeightSetBase_h
#define __itkWeightSetBase_h

#include "itkLightProcessObject.h"
#include <vnl/vnl_matrix.h>
#include <vnl/vnl_diag_matrix.h>
#include "itkMacro.h"
#include "itkVector.h"
#include "itkMersenneTwisterRandomVariateGenerator.h"
#include <math.h>
#include <stdlib.h>

namespace itk
{
  namespace Statistics
    {

    template<class TMeasurementVector, class TTargetVector>
      class WeightSetBase : public LightProcessObject
        {
      public:

        typedef WeightSetBase            Self;
        typedef LightProcessObject       Superclass;
        typedef SmartPointer<Self>       Pointer;
        typedef SmartPointer<const Self> ConstPointer;
        itkTypeMacro(WeightSetBase, LightProcessObject);

        typedef MersenneTwisterRandomVariateGenerator  RandomVariateGeneratorType;

        typedef typename TMeasurementVector::ValueType ValueType;
        typedef ValueType*                  ValuePointer;
        typedef const ValueType*            ValueConstPointer;

        void Initialize();

        ValueType RandomWeightValue(ValueType low, ValueType high);

        virtual void ForwardPropagate(ValuePointer inputlayeroutputvalues);
        virtual void BackwardPropagate(ValuePointer inputerror);

        void SetConnectivityMatrix(vnl_matrix < int>);

        void SetNumberOfInputNodes(unsigned int n);
        unsigned int GetNumberOfInputNodes() const;

        void SetNumberOfOutputNodes(unsigned int n);
        unsigned int GetNumberOfOutputNodes() const;

        void SetRange(ValueType Range);

        virtual ValuePointer GetOutputValues();
        virtual ValuePointer GetInputValues();

        ValuePointer GetTotalDeltaValues();

        ValuePointer GetTotalDeltaBValues();

        ValuePointer GetDeltaValues();

        void SetDeltaValues(ValuePointer);

        void SetDWValues(ValuePointer);

        void SetDBValues(ValuePointer);

        ValuePointer GetDeltaBValues();

        void SetDeltaBValues(ValuePointer);

        ValuePointer GetDWValues();

        ValuePointer GetPrevDWValues();

        ValuePointer GetPrevDBValues();

        ValuePointer GetPrev_m_2DWValues();

        ValuePointer GetPrevDeltaValues();

        ValuePointer GetPrev_m_2DeltaValues();

        ValuePointer GetPrevDeltaBValues();

        ValuePointer GetWeightValues();
        ValueConstPointer GetWeightValues() const;


        void SetWeightValues(ValuePointer weights);

        virtual void UpdateWeights(ValueType LearningRate);

        itkSetMacro( Momentum, ValueType );
        itkGetConstReferenceMacro( Momentum, ValueType );

        itkSetMacro( Bias, ValueType );
        itkGetConstReferenceMacro( Bias, ValueType );

        itkSetMacro( FirstPass, bool );
        itkGetConstMacro( FirstPass, bool );

        itkSetMacro( SecondPass, bool );
        itkGetConstMacro( SecondPass, bool );

        void InitializeWeights();

        itkSetMacro(WeightSetId,unsigned int);
        itkGetConstMacro(WeightSetId,unsigned int);

        itkSetMacro(InputLayerId,unsigned int);
        itkGetConstMacro(InputLayerId,unsigned int);

        itkSetMacro(OutputLayerId,unsigned int);
        itkGetConstMacro(OutputLayerId,unsigned int);

      protected:

        WeightSetBase();
        ~WeightSetBase();

        /** Method to print the object. */
        virtual void PrintSelf( std::ostream& os, Indent indent ) const;

        typename RandomVariateGeneratorType::Pointer m_RandomGenerator;

        unsigned int              m_NumberOfInputNodes;
        unsigned int              m_NumberOfOutputNodes;
        vnl_matrix<ValueType>     m_OutputValues;
        vnl_matrix<ValueType>     m_InputErrorValues;

        // weight updates dw=lr * del *y
        // DW= current
        // DW_m_1 = previous
        // DW_m_2= second to last
        // same applies for delta and bias values

        vnl_matrix<ValueType> m_DW;            // delta valies for weight update
        vnl_matrix<ValueType> m_DW_new;        // delta valies for weight update
        vnl_matrix<ValueType> m_DW_m_1;        // delta valies for weight update
        vnl_matrix<ValueType> m_DW_m_2;        // delta valies for weight update
        vnl_matrix<ValueType> m_DW_m;          // delta valies for weight update

        vnl_vector<ValueType> m_DB;            // delta values for bias update
        vnl_vector<ValueType> m_DB_new;        // delta values for bias update
        vnl_vector<ValueType> m_DB_m_1;        // delta values for bias update
        vnl_vector<ValueType> m_DB_m_2;        // delta values for bias update

        vnl_matrix<ValueType> m_Del;           // dw=lr * del * y
        vnl_matrix<ValueType> m_Del_new;       // dw=lr * del * y
        vnl_matrix<ValueType> m_Del_m_1;       // dw=lr * del * y
        vnl_matrix<ValueType> m_Del_m_2;       // dw=lr * del * y

        vnl_vector<ValueType> m_Delb;          // delta values for bias update
        vnl_vector<ValueType> m_Delb_new;      // delta values for bias update
        vnl_vector<ValueType> m_Delb_m_1;      // delta values for bias update
        vnl_vector<ValueType> m_Delb_m_2;      // delta values for bias update

        vnl_matrix<ValueType> m_InputLayerOutput;
        vnl_matrix<ValueType> m_WeightMatrix;  // composed of weights and a column
        // of biases
        vnl_matrix<int>       m_ConnectivityMatrix;

        ValueType m_Momentum;
        ValueType m_Bias;
        bool      m_FirstPass;
        bool      m_SecondPass;
        ValueType m_Range;

        unsigned int m_InputLayerId;
        unsigned int m_OutputLayerId;
        unsigned int m_WeightSetId;

        };  //class

    } // end namespace Statistics
} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkWeightSetBase.txx"
#endif


#endif