123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127 |
- #ifndef _INCLUDEFEATURELEARNINGPROTOTYPES
- #define _INCLUDEFEATURELEARNINGPROTOTYPES
- #include "FeatureLearningGeneric.h"
- #include <core/vector/Distance.h>
- #include <core/vector/VVector.h>
- #include <vislearning/cbaselib/MultiDataset.h>
- #include <vislearning/features/localfeatures/GenericLocalFeatureSelection.h>
- #include <vislearning/features/simplefeatures/CodebookPrototypes.h>
- #include <vislearning/math/cluster/ClusterAlgorithm.h>
- namespace OBJREC
- {
-
-
-
- class FeatureLearningPrototypes : public FeatureLearningGeneric
- {
- protected:
-
-
-
-
- bool showTrainingImages;
-
-
- int initialNumberOfClusters;
-
- OBJREC::ClusterAlgorithm * clusterAlgo;
-
- NICE::VectorDistance<double> * distFunction;
- LocalFeatureRepresentation * featureExtractor;
-
-
- OBJREC::CodebookPrototypes prototypes;
-
- int newImageCounter;
-
- double oldMaxDist;
-
-
- double maxValForVisualization;
-
-
- int i_posXOfNewPrototype;
- int i_posYOfNewPrototype;
-
-
-
-
- void setClusterAlgo( const std::string & _clusterAlgoString);
- void setFeatureExtractor( const bool & _setForTraining = false);
-
- void extractFeaturesFromTrainingImages( const OBJREC::MultiDataset *_md, NICE::VVector & examplesTraining );
-
- void train ( const OBJREC::MultiDataset *_md );
-
- virtual bool loadInitialCodebook ( );
- virtual bool writeInitialCodebook ( );
-
- virtual void evaluateCurrentCodebookForGivenFeatures ( const NICE::VVector & _features,
- const NICE::VVector & _positions,
- NICE::FloatImage & _noveltyImageGaussFiltered,
- NICE::FloatImage * _noveltyImage = NULL
- );
- public:
-
- FeatureLearningPrototypes ( const NICE::Config *_conf, const OBJREC::MultiDataset *_md , const std::string & _section = "featureLearning");
-
- virtual ~FeatureLearningPrototypes();
-
-
- virtual void learnNewFeatures ( const std::string & _filename ) = 0 ;
-
- virtual NICE::FloatImage evaluateCurrentCodebookByDistance ( const std::string & _filename , const bool & beforeComputingNewFeatures = true );
-
- virtual NICE::ImageT<int> evaluateCurrentCodebookByAssignments ( const std::string & _filename , const bool & beforeComputingNewFeatures = true, const bool & _binaryShowLatestPrototype = false);
-
- virtual void evaluateCurrentCodebookByConfusionMatrix( NICE::Matrix & _confusionMat );
-
- virtual NICE::VVector * getCurrentCodebook();
-
- };
- }
- #endif
|