itkNeuralNetworkFileReader.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkNeuralNetworkFileReader.h,v $
  Language:  C++
  Date:      $Date: 2007/02/13 19:03:13 $
  Version:   $Revision: 1.8 $

  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 __itkNeuralNetworkFileReader_h
#define __itkNeuralNetworkFileReader_h

#include "metaTypes.h"
#include "metaUtils.h"

#include "itkMultilayerNeuralNetworkBase.h"

#include "itkBackPropagationLayer.h"

#include "itkCompletelyConnectedWeightSet.h"

#include "itkSumInputFunction.h"
#include "itkProductInputFunction.h"

#include "itkIdentityTransferFunction.h"
#include "itkLogSigmoidTransferFunction.h"
#include "itkSigmoidTransferFunction.h"
#include "itkTanSigmoidTransferFunction.h"
#include "itkSymmetricSigmoidTransferFunction.h"


namespace itk
{

template<class TVector, class TOutput>
class NeuralNetworkFileReader : public Object
{
public:
  
  typedef NeuralNetworkFileReader                       Self;
  typedef Object                                        Superclass;
  typedef SmartPointer<Self>                            Pointer;
  typedef SmartPointer<const Self>                      ConstPointer;

  itkTypeMacro(NeuralNetworkFileReader,Object);

  itkNewMacro(Self);

  typedef Statistics::MultilayerNeuralNetworkBase<TVector,TOutput> NetworkType;
  typedef typename NetworkType::Pointer NetworkPointer;
  
  typedef Statistics::LayerBase<TVector,TOutput>        LayerType;
  typedef typename LayerType::Pointer                   LayerPointer;
  
  typedef typename LayerType::ValueType ValueType;

  typedef typename NetworkType::WeightSetType WeightSetType;
  typedef Statistics::CompletelyConnectedWeightSet<TVector, TOutput> 
                                          CompletelyConnectedWeighttype;
 
  typedef Statistics::BackPropagationLayer<TVector,TOutput> BackPropLayerType;
  typedef typename BackPropLayerType::Pointer               BPLayerPointerType;
   
  typedef typename CompletelyConnectedWeighttype::Pointer 
                                          CompletelyWeightSetPointer;

  typedef typename WeightSetType::Pointer WeightSetPointer;
 
  itkSetStringMacro( FileName );

  itkGetStringMacro( FileName );

  void Update(void);

  NetworkType * GetOutput() const;

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

 
protected:
  NeuralNetworkFileReader();
  virtual ~NeuralNetworkFileReader();
  virtual void PrintSelf( std::ostream& os, Indent indent ) const;
  
private:

  void ClearFields();
  typedef std::vector<MET_FieldRecordType *>      FieldsContainerType;
  typedef std::vector< LayerPointer >             LayersContainer;
  typedef std::vector<WeightSetPointer>           WeightsContainer;

  typedef struct 
    { 
    std::string name;
    std::string value;
    }                                              LineType;

  typedef std::list< LineType >                    LinesContainer;


  NetworkPointer         m_Network;
  int                    m_ReadWeightValuesType;
  LayersContainer        m_Layers;
  WeightsContainer       m_Weights;
  LinesContainer         m_NameValue;  
  std::string            m_FileName;
  FieldsContainerType    m_Fields;
  std::ifstream          m_InputFile;

};

} // namespace itk

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

#endif 

---