123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205 |
- #ifndef SEGMENTATIONMODULE_HPP_
- #define SEGMENTATIONMODULE_HPP_
- #include "opencv2/core.hpp"
- namespace cv
- {
- namespace bioinspired
- {
- struct SegmentationParameters{
-
- SegmentationParameters():
- thresholdON(100),
- thresholdOFF(100),
- localEnergy_temporalConstant(0.5),
- localEnergy_spatialConstant(5),
- neighborhoodEnergy_temporalConstant(1),
- neighborhoodEnergy_spatialConstant(15),
- contextEnergy_temporalConstant(1),
- contextEnergy_spatialConstant(75){};
-
- float thresholdON;
- float thresholdOFF;
-
- float localEnergy_temporalConstant;
-
- float localEnergy_spatialConstant;
-
- float neighborhoodEnergy_temporalConstant;
- float neighborhoodEnergy_spatialConstant;
-
- float contextEnergy_temporalConstant;
- float contextEnergy_spatialConstant;
- };
- class CV_EXPORTS_W TransientAreasSegmentationModule: public Algorithm
- {
- public:
-
- CV_WRAP virtual Size getSize()=0;
-
- CV_WRAP virtual void setup(String segmentationParameterFile="", const bool applyDefaultSetupOnFailure=true)=0;
-
- virtual void setup(cv::FileStorage &fs, const bool applyDefaultSetupOnFailure=true)=0;
-
- virtual void setup(SegmentationParameters newParameters)=0;
-
- virtual SegmentationParameters getParameters()=0;
-
- CV_WRAP virtual const String printSetup()=0;
-
- CV_WRAP virtual void write( String fs ) const=0;
-
- virtual void write( cv::FileStorage& fs ) const=0;
-
- CV_WRAP virtual void run(InputArray inputToSegment, const int channelIndex=0)=0;
-
- CV_WRAP virtual void getSegmentationPicture(OutputArray transientAreas)=0;
-
- CV_WRAP virtual void clearAllBuffers()=0;
- };
- CV_EXPORTS_W Ptr<TransientAreasSegmentationModule> createTransientAreasSegmentationModule(Size inputSize);
- }}
- #endif
|