itkNeuralNetworkFileWriter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkNeuralNetworkFileWriter.h,v $
  Language:  C++
  Date:      $Date: 2007/02/13 21:37:03 $
  Version:   $Revision: 1.7 $

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

#include <metaTypes.h>
#include <metaUtils.h>
#include <typeinfo>

#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 NeuralNetworkFileWriter : public Object
{
public:
  
  typedef NeuralNetworkFileWriter           Self;
  typedef Object                            Superclass;
  typedef SmartPointer<Self>                Pointer;
  typedef SmartPointer<const Self>          ConstPointer;

  itkTypeMacro(NeuralNetworkFileWriter,Object);

  itkNewMacro(Self);

  typedef Statistics::MultilayerNeuralNetworkBase<TVector,TOutput> NetworkType;

  typedef typename NetworkType::Pointer        NetworkPointer;
  typedef typename NetworkType::ConstPointer   NetworkConstPointer;
  
  typedef Statistics::LayerBase<TVector,TOutput>                  LayerType;

  typedef typename LayerType::Pointer                      LayerPointer;
  typedef typename LayerType::ConstPointer                 LayerConstPointer;

  typedef typename LayerType::TransferFunctionPointer TransferFunctionPointer;

  typedef typename LayerType::TransferFunctionConstPointer 
                                              TransferFunctionConstPointer;

  typedef typename LayerType::InputFunctionPointer InputFunctionPointer;

  typedef typename LayerType::InputFunctionConstPointer
                                                      InputFunctionConstPointer;

  typedef typename LayerType::WeightSetPointer        WeightSetPointer;
  typedef typename LayerType::WeightSetConstPointer   WeightSetConstPointer;
  typedef typename LayerType::ValueType               ValueType;

  typedef typename NetworkType::WeightSetType               WeightSetType;
  typedef Statistics::BackPropagationLayer<TVector,TOutput> BackPropLayerType;
  typedef typename BackPropLayerType::Pointer               BPLayerPointerType;
  
  itkSetStringMacro(FileName);

  itkGetStringMacro(FileName);

  void SetInput( const NetworkType* network );
  const NetworkType * GetInput() const;

  void Update(void);

  itkSetMacro(WriteWeightValuesType, unsigned int);
  itkGetConstMacro(WriteWeightValuesType, unsigned int);
  
protected:
  NeuralNetworkFileWriter();
  virtual ~NeuralNetworkFileWriter();
  virtual void PrintSelf( std::ostream& os, Indent indent ) const;

private:
  void ClearFields();
  typedef std::vector<MET_FieldRecordType *> FieldsContainerType;

  NetworkConstPointer    m_Network;
  unsigned int           m_WriteWeightValuesType;
  
  std::string            m_FileName;
  FieldsContainerType    m_Fields;

  std::ofstream          m_OutputFile;
};

} // namespace itk

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

#endif 

---