00001 /*!@file Gist/trainUtils.H training utility functions (not limited to FFN) */ 00002 00003 // //////////////////////////////////////////////////////////////////// // 00004 // The iLab Neuromorphic Vision C++ Toolkit - Copyright (C) 2001 by the // 00005 // University of Southern California (USC) and the iLab at USC. // 00006 // See http://iLab.usc.edu for information about this project. // 00007 // //////////////////////////////////////////////////////////////////// // 00008 // Major portions of the iLab Neuromorphic Vision Toolkit are protected // 00009 // under the U.S. patent ``Computation of Intrinsic Perceptual Saliency // 00010 // in Visual Environments, and Applications'' by Christof Koch and // 00011 // Laurent Itti, California Institute of Technology, 2001 (patent // 00012 // pending; application number 09/912,225 filed July 23, 2001; see // 00013 // http://pair.uspto.gov/cgi-bin/final/home.pl for current status). // 00014 // //////////////////////////////////////////////////////////////////// // 00015 // This file is part of the iLab Neuromorphic Vision C++ Toolkit. // 00016 // // 00017 // The iLab Neuromorphic Vision C++ Toolkit is free software; you can // 00018 // redistribute it and/or modify it under the terms of the GNU General // 00019 // Public License as published by the Free Software Foundation; either // 00020 // version 2 of the License, or (at your option) any later version. // 00021 // // 00022 // The iLab Neuromorphic Vision C++ Toolkit is distributed in the hope // 00023 // that it will be useful, but WITHOUT ANY WARRANTY; without even the // 00024 // implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR // 00025 // PURPOSE. See the GNU General Public License for more details. // 00026 // // 00027 // You should have received a copy of the GNU General Public License // 00028 // along with the iLab Neuromorphic Vision C++ Toolkit; if not, write // 00029 // to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, // 00030 // Boston, MA 02111-1307 USA. // 00031 // //////////////////////////////////////////////////////////////////// // 00032 // 00033 // Primary maintainer for this file: Christian Siagian <siagian@usc.edu> 00034 // $HeadURL: svn://isvn.usc.edu/software/invt/trunk/saliency/src/Gist/trainUtils.H $ 00035 // $Id: trainUtils.H 8267 2007-04-18 18:24:24Z rjpeters $ 00036 // 00037 00038 #ifndef GIST_TRAINUTILS_H_DEFINED 00039 #define GIST_TRAINUTILS_H_DEFINED 00040 00041 #include "Image/Image.H" 00042 #include <string> 00043 00044 template <class T> class Image; 00045 00046 // ###################################################################### 00047 // utilities for training feed forward network (FFN) 00048 00049 // ###################################################################### 00050 // to store information from the training file 00051 class FFNtrainInfo 00052 { 00053 public: 00054 //! Construct a FFN training params info 00055 //! if blank, need to call reset later on 00056 FFNtrainInfo(std::string fName = std::string("")); 00057 00058 //! Destructor 00059 virtual ~FFNtrainInfo(); 00060 00061 //! reset the Neural Network classifier parameter 00062 void reset(std::string fName); 00063 00064 std::string trainFolder; //!< where the training data is 00065 std::string testFolder; //!< where the testing data is 00066 uint nOutput; //!< the number of output dimension 00067 bool isPCA; //!< has a dimension reduction step 00068 std::string evecFname; //!< the dimension reduction 00069 uint oriFeatSize; //!< the original number of features 00070 uint redFeatSize; //!< the reduced number of features 00071 uint h1size; //!< the number of hidden nodes in layer 1 00072 uint h2size; //!< the number of hidden nodes in layer 2 00073 std::string h1Name; //!< the weight file for hidden layer 1 00074 std::string h2Name; //!< the weight file for hidden layer 2 00075 std::string oName; //!< the weight file for output layer 00076 float learnRate; //!< the learning rate 00077 std::string trainSampleFile; //!< the training sample file list 00078 std::string testSampleFile; //!< the testing sample file list 00079 }; 00080 00081 // ###################################################################### 00082 // utilities for applying PCA/ICA dimension reduction 00083 00084 //! Setup the PCA/ICA unmixing matrix 00085 Image<double> setupPcaIcaMatrix(std::string inW, int oriSize, int redSize); 00086 00087 //! Get the PCA/ICA reduced feature vector 00088 //! in a form of an image 00089 Image<float> getPcaIcaFeatImage(Image<double> res, int w, int h, int s); 00090 00091 #endif // GIST_TRAINUTILS_H_DEFINED
1.4.4