123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135 |
- #ifndef RELATIVELOCATIONPRIORINCLUDE
- #define RELATIVELOCATIONPRIORINCLUDE
- #include "core/image/MultiChannelImageT.h"
- #include "vislearning/cbaselib/CachedExample.h"
- #include "vislearning/baselib/Preprocess.h"
- #include "vislearning/baselib/Globals.h"
- #include "vislearning/classifier/fpclassifier/logisticregression/SLR.h"
- #include "vislearning/classifier/fpclassifier/randomforest/FPCRandomForests.h"
- #include "vislearning/features/fpfeatures/SparseVectorFeature.h"
- #include "vislearning/cbaselib/ClassNames.h"
- namespace OBJREC
- {
- class RelativeLocationPrior : public NICE::Persistent
- {
- protected:
-
- std::vector<NICE::MultiChannelImageT<double> *> priormaps;
-
- const NICE::Config *conf;
-
- int classno;
-
- int mapsize;
-
- void convertCoords ( int &x, int xsize );
-
- Examples trainingsdata;
-
- std::vector<SLR> classifiers;
-
- int featdim;
- public:
-
- RelativeLocationPrior();
-
- RelativeLocationPrior ( const NICE::Config *_conf );
-
- ~RelativeLocationPrior();
-
- void setClassNo ( int _classno );
-
- void Init();
-
- void trainPriorsMaps ( Examples ®ions, int xsize, int ysize );
-
- void finishPriorsMaps ( ClassNames &cn );
-
- void trainClassifier ( Examples ®ions, NICE::MultiChannelImageT<double> & probabilities );
-
- void finishClassifier();
-
- void getFeature ( Examples ®ions, NICE::MultiChannelImageT<double> & probabilities );
-
- void postprocess ( Examples ®ions, NICE::MultiChannelImageT<double> & probabilities );
-
- void restore ( std::istream & is, int format = 0 );
-
- void store ( std::ostream & os, int format = 0 ) const;
-
- void clear ();
- };
- }
- #endif
|