Skip to content

O
opencv_contrib
Commit 15c4d950 authored by Alfonso Sanchez-Beato's avatar Alfonso Sanchez-Beato
Browse files
Options

Merge branch 'master' of https://github.com/Itseez/opencv_contrib

parents 25e39c69 319ee932
Hide whitespace changes
Inline Side-by-side
Showing with 807 additions and 1 deletion
README.md
View file @ 15c4d950
......@@ -29,3 +29,19 @@ use CMake's `BUILD_opencv_*` options. Like in this example:
```
$ cmake -DOPENCV_EXTRA_MODULES_PATH=<opencv_contrib>/modules -DBUILD_opencv_legacy=OFF <opencv_source_directory>
```
If you prefer using the gui version of cmake (cmake-gui), then, you can add `opencv_contrib` modules within `opencv` core by doing the following:
1. start cmake-gui
2. select the opencv source code folder and the folder where binaries will be built (the 2 upper forms of the interface)
3. press the `configure` button. you will see all the opencv build parameters in the central interface
4. browse the parameters and look for the form called `OPENCV_EXTRA_MODULES_PATH` (use the search form to focus rapidly on it)
5. complete this `OPENCV_EXTRA_MODULES_PATH` by the proper pathname to the `<opencv_contrib>/modules` value using its browse button.
6. press the `configure` button followed by the `generate` button (the first time, you will be asked which makefile style to use)
7. build the `opencv` core with the method you chose (make and make install if you chose Unix makfile at step 6)
modules/bioinspired/include/opencv2/bioinspired.hpp
View file @ 15c4d950
......@@ -46,5 +46,5 @@
#include "opencv2/core.hpp"
#include "opencv2/bioinspired/retina.hpp"
#include "opencv2/bioinspired/retinafasttonemapping.hpp"
#include "opencv2/bioinspired/transientareassegmentationmodule.hpp"
#endif
modules/bioinspired/include/opencv2/bioinspired/transientareassegmentationmodule.hpp 0 → 100755
View file @ 15c4d950
/*#******************************************************************************
** IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
**
** By downloading, copying, installing or using the software you agree to this license.
** If you do not agree to this license, do not download, install,
** copy or use the software.
**
**
** bioinspired : interfaces allowing OpenCV users to integrate Human Vision System models.
** TransientAreasSegmentationModule Use: extract areas that present spatio-temporal changes.
** => It should be used at the output of the cv::bioinspired::Retina::getMagnoRAW() output that enhances spatio-temporal changes
**
** Maintainers : Listic lab (code author current affiliation & applications)
**
** Creation - enhancement process 2007-2013
** Author: Alexandre Benoit (benoit.alexandre.vision@gmail.com), LISTIC lab, Annecy le vieux, France
**
** Theses algorithm have been developped by Alexandre BENOIT since his thesis with Alice Caplier at Gipsa-Lab (www.gipsa-lab.inpg.fr) and the research he pursues at LISTIC Lab (www.listic.univ-savoie.fr).
** Refer to the following research paper for more information:
** Strat S. T. , Benoit A.Lambert P. , Caplier A., "Retina Enhanced SURF Descriptors for Spatio-Temporal Concept Detection", Multimedia Tools and Applications, 2012 (DOI: 10.1007/s11042-012-1280-0)
** Benoit A., Caplier A., Durette B., Herault, J., "USING HUMAN VISUAL SYSTEM MODELING FOR BIO-INSPIRED LOW LEVEL IMAGE PROCESSING", Elsevier, Computer Vision and Image Understanding 114 (2010), pp. 758-773, DOI: http://dx.doi.org/10.1016/j.cviu.2010.01.011
** This work have been carried out thanks to Jeanny Herault who's research and great discussions are the basis of all this work, please take a look at his book:
** Vision: Images, Signals and Neural Networks: Models of Neural Processing in Visual Perception (Progress in Neural Processing),By: Jeanny Herault, ISBN: 9814273686. WAPI (Tower ID): 113266891.
**
**
** License Agreement
** For Open Source Computer Vision Library
**
** Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
** Copyright (C) 2008-2011, Willow Garage Inc., all rights reserved.
**
** For Human Visual System tools (bioinspired)
** Copyright (C) 2007-2011, LISTIC Lab, Annecy le Vieux and GIPSA Lab, Grenoble, France, all rights reserved.
**
** Third party copyrights are property of their respective owners.
**
** Redistribution and use in source and binary forms, with or without modification,
** are permitted provided that the following conditions are met:
**
** * Redistributions of source code must retain the above copyright notice,
** this list of conditions and the following disclaimer.
**
** * Redistributions in binary form must reproduce the above copyright notice,
** this list of conditions and the following disclaimer in the documentation
** and/or other materials provided with the distribution.
**
** * The name of the copyright holders may not be used to endorse or promote products
** derived from this software without specific prior written permission.
**
** This software is provided by the copyright holders and contributors "as is" and
** any express or implied warranties, including, but not limited to, the implied
** warranties of merchantability and fitness for a particular purpose are disclaimed.
** In no event shall the Intel Corporation or contributors be liable for any direct,
** indirect, incidental, special, exemplary, or consequential damages
** (including, but not limited to, procurement of substitute goods or services;
** loss of use, data, or profits; or business interruption) however caused
** and on any theory of liability, whether in contract, strict liability,
** or tort (including negligence or otherwise) arising in any way out of
** the use of this software, even if advised of the possibility of such damage.
*******************************************************************************/
#ifndef SEGMENTATIONMODULE_HPP_
#define SEGMENTATIONMODULE_HPP_
/**
* @class TransientAreasSegmentationModule
* @brief class which provides a transient/moving areas segmentation module
* -> perform a locally adapted segmentation by using the retina magno input data
* @author Alexandre BENOIT, benoit.alexandre.vision@gmail.com
* Release date 2007-2013
* Based on Alexandre BENOIT thesis: "Le système visuel humain au secours de la vision par ordinateur"
* -> 3 spatio temporal filters are used:
* a first one which filters the noise and local variations of the input motion energy
* a second (more powerfull low pass spatial filter) which gives the neighborhood motion energy
* -> the segmentation consists in the comparison of these both outputs, if the local motion energy is higher to the neighborhood otion energy, then the area is considered as moving and is segmented
* => a stronger third low pass filter helps decision by providing a smooth information about the "motion context" in a wider area
*/
#include "opencv2/core.hpp" // for all OpenCV core functionalities access, including cv::Exception support
namespace cv
{
namespace bioinspired
{
class CV_EXPORTS TransientAreasSegmentationModule: public Algorithm
{
public:
// parameters structure
struct SegmentationParameters{
SegmentationParameters():
thresholdON(100),
thresholdOFF(100),
// local energy filtering parameters : the aim is to smooth local textures and residual noise
localEnergy_temporalConstant(0.5), // the time constant of the first order low pass filter, use it to cut high temporal frequencies (noise or fast motion), unit is frames, typical value is 0.5 frame
localEnergy_spatialConstant(5), // the spatial constant of the first order low pass filter, use it to cut high spatial frequencies (noise or thick contours), unit is pixels, typical value is 5 pixel
// local neighborhood energy filtering parameters : the aim is to get information about the energy neighborhood to perform a center surround energy analysis
neighborhoodEnergy_temporalConstant(1),
neighborhoodEnergy_spatialConstant(15),
// context neighborhood energy filtering parameters : the aim is to get information about the energy on a wide neighborhood area to filtered out local effects
contextEnergy_temporalConstant(1),
contextEnergy_spatialConstant(75){};// default setup
float thresholdON, thresholdOFF;
float localEnergy_temporalConstant, localEnergy_spatialConstant;
float neighborhoodEnergy_temporalConstant, neighborhoodEnergy_spatialConstant;
float contextEnergy_temporalConstant, contextEnergy_spatialConstant;
};
/**
* @return the sze of the manage input and output images
*/
virtual Size getSize()=0;
/**
* try to open an XML segmentation parameters file to adjust current segmentation instance setup
* => if the xml file does not exist, then default setup is applied
* => warning, Exceptions are thrown if read XML file is not valid
* @param retinaParameterFile : the parameters filename
* @param applyDefaultSetupOnFailure : set to true if an error must be thrown on error
*/
virtual void setup(String segmentationParameterFile="", const bool applyDefaultSetupOnFailure=true)=0;
/**
* try to open an XML segmentation parameters file to adjust current segmentation instance setup
* => if the xml file does not exist, then default setup is applied
* => warning, Exceptions are thrown if read XML file is not valid
* @param fs : the open Filestorage which contains segmentation parameters
* @param applyDefaultSetupOnFailure : set to true if an error must be thrown on error
*/
virtual void setup(cv::FileStorage &fs, const bool applyDefaultSetupOnFailure=true)=0;
/**
* try to open an XML segmentation parameters file to adjust current segmentation instance setup
* => if the xml file does not exist, then default setup is applied
* => warning, Exceptions are thrown if read XML file is not valid
* @param newParameters : a parameters structures updated with the new target configuration
* @param applyDefaultSetupOnFailure : set to true if an error must be thrown on error
*/
virtual void setup(TransientAreasSegmentationModule::SegmentationParameters newParameters)=0;
/**
* @return the current parameters setup
*/
virtual struct TransientAreasSegmentationModule::TransientAreasSegmentationModule::SegmentationParameters getParameters()=0;
/**
* parameters setup display method
* @return a string which contains formatted parameters information
*/
virtual const String printSetup()=0;
/**
* write xml/yml formated parameters information
* @rparam fs : the filename of the xml file that will be open and writen with formatted parameters information
*/
virtual void write( String fs ) const=0;
/**
* write xml/yml formated parameters information
* @param fs : a cv::Filestorage object ready to be filled
*/
virtual void write( cv::FileStorage& fs ) const=0;
/**
* main processing method, get result using methods getSegmentationPicture()
* @param inputToSegment : the image to process, it must match the instance buffer size !
* @param channelIndex : the channel to process in case of multichannel images
*/
virtual void run(InputArray inputToSegment, const int channelIndex=0)=0;
/**
* access function
* @return the last segmentation result: a boolean picture which is resampled between 0 and 255 for a display purpose
*/
virtual void getSegmentationPicture(OutputArray transientAreas)=0;
/**
* cleans all the buffers of the instance
*/
virtual void clearAllBuffers()=0;
};
/**
* allocator
* @param Size : size of the images input to segment (output will be the same size)
*/
CV_EXPORTS Ptr<TransientAreasSegmentationModule> createTransientAreasSegmentationModule(Size inputSize);
}} // namespaces end : cv and bioinspired
#endif
modules/bioinspired/src/transientareassegmentationmodule.cpp 0 → 100644
View file @ 15c4d950
/*#******************************************************************************
** IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
**
** By downloading, copying, installing or using the software you agree to this license.
** If you do not agree to this license, do not download, install,
** copy or use the software.
**
**
** bioinspired : interfaces allowing OpenCV users to integrate Human Vision System models.
** TransientAreasSegmentationModule Use: extract areas that present spatio-temporal changes.
** => It should be used at the output of the cv::bioinspired::Retina::getMagnoRAW() output that enhances spatio-temporal changes
**
** Maintainers : Listic lab (code author current affiliation & applications)
**
** Creation - enhancement process 2007-2013
** Author: Alexandre Benoit (benoit.alexandre.vision@gmail.com), LISTIC lab, Annecy le vieux, France
**
** Theses algorithm have been developped by Alexandre BENOIT since his thesis with Alice Caplier at Gipsa-Lab (www.gipsa-lab.inpg.fr) and the research he pursues at LISTIC Lab (www.listic.univ-savoie.fr).
** Refer to the following research paper for more information:
** Strat S. T. , Benoit A.Lambert P. , Caplier A., "Retina Enhanced SURF Descriptors for Spatio-Temporal Concept Detection", Multimedia Tools and Applications, 2012 (DOI: 10.1007/s11042-012-1280-0)
** Benoit A., Caplier A., Durette B., Herault, J., "USING HUMAN VISUAL SYSTEM MODELING FOR BIO-INSPIRED LOW LEVEL IMAGE PROCESSING", Elsevier, Computer Vision and Image Understanding 114 (2010), pp. 758-773, DOI: http://dx.doi.org/10.1016/j.cviu.2010.01.011
** This work have been carried out thanks to Jeanny Herault who's research and great discussions are the basis of all this work, please take a look at his book:
** Vision: Images, Signals and Neural Networks: Models of Neural Processing in Visual Perception (Progress in Neural Processing),By: Jeanny Herault, ISBN: 9814273686. WAPI (Tower ID): 113266891.
**
**
** License Agreement
** For Open Source Computer Vision Library
**
** Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
** Copyright (C) 2008-2011, Willow Garage Inc., all rights reserved.
**
** For Human Visual System tools (bioinspired)
** Copyright (C) 2007-2011, LISTIC Lab, Annecy le Vieux and GIPSA Lab, Grenoble, France, all rights reserved.
**
** Third party copyrights are property of their respective owners.
**
** Redistribution and use in source and binary forms, with or without modification,
** are permitted provided that the following conditions are met:
**
** * Redistributions of source code must retain the above copyright notice,
** this list of conditions and the following disclaimer.
**
** * Redistributions in binary form must reproduce the above copyright notice,
** this list of conditions and the following disclaimer in the documentation
** and/or other materials provided with the distribution.
**
** * The name of the copyright holders may not be used to endorse or promote products
** derived from this software without specific prior written permission.
**
** This software is provided by the copyright holders and contributors "as is" and
** any express or implied warranties, including, but not limited to, the implied
** warranties of merchantability and fitness for a particular purpose are disclaimed.
** In no event shall the Intel Corporation or contributors be liable for any direct,
** indirect, incidental, special, exemplary, or consequential damages
** (including, but not limited to, procurement of substitute goods or services;
** loss of use, data, or profits; or business interruption) however caused
** and on any theory of liability, whether in contract, strict liability,
** or tort (including negligence or otherwise) arising in any way out of
** the use of this software, even if advised of the possibility of such damage.
*******************************************************************************/
/**
* @class TransientAreasSegmentationModule
* @brief class which provides a transient/moving areas segmentation module
* -> perform a locally adapted segmentation by using the retina magno output data
* @author Alexandre BENOIT, benoit.alexandre.vision@gmail.com
* Release date 2007-2014
* Based on Alexandre BENOIT thesis: "Le système visuel humain au secours de la vision par ordinateur"
* -> 3 spatio temporal filters are used:
* a first one which filters the noise and local variations of the input motion energy
* a second (more powerfull low pass spatial filter) which gives the neighborhood motion energy
* a third that measures the global motion context on a wide area
* -> the segmentation consists in the comparison of these filter outputs, if the local motion energy is higher to the neighborhood motion energy, then the area is considered as moving and is segmented
*/
#include "precomp.hpp"
#include "basicretinafilter.hpp"
#define _SEGMENTATIONDEBUG //define SEGMENTATIONDEBUG to access more data/methods
namespace cv
{
namespace bioinspired
{
class TransientAreasSegmentationModuleImpl: protected BasicRetinaFilter
{
public:
/**
* constructor
* @param Size : size of the images input to segment (output will be the same size)
*/
TransientAreasSegmentationModuleImpl(const Size inputSize);
/**
* standard destructor
*/
virtual ~TransientAreasSegmentationModuleImpl();
/**
* @return the size of the manage input and output images
*/
Size getSize(){return cv::Size(getNBcolumns(), getNBrows());};
/**
* try to open an XML segmentation parameters file to adjust current segmentation instance setup
* => if the xml file does not exist, then default setup is applied
* => warning, Exceptions are thrown if read XML file is not valid
* @param retinaParameterFile : the parameters filename
* @param applyDefaultSetupOnFailure : set to true if an error must be thrown on error
*/
void setup(String segmentationParameterFile="", const bool applyDefaultSetupOnFailure=true);
/**
* try to open an XML segmentation parameters file to adjust current segmentation instance setup
* => if the xml file does not exist, then default setup is applied
* => warning, Exceptions are thrown if read XML file is not valid
* @param fs : the open Filestorage which contains segmentation parameters
* @param applyDefaultSetupOnFailure : set to true if an error must be thrown on error
*/
void setup(cv::FileStorage &fs, const bool applyDefaultSetupOnFailure=true);
/**
* try to open an XML segmentation parameters file to adjust current segmentation instance setup
* => if the xml file does not exist, then default setup is applied
* => warning, Exceptions are thrown if read XML file is not valid
* @param newParameters : a parameters structures updated with the new target configuration
* @param applyDefaultSetupOnFailure : set to true if an error must be thrown on error
*/
void setup(TransientAreasSegmentationModule::SegmentationParameters newParameters);
/**
* @return the current parameters setup
*/
struct TransientAreasSegmentationModule::TransientAreasSegmentationModule::SegmentationParameters getParameters();
/**
* parameters setup display method
* @return a string which contains formatted parameters information
*/
const String printSetup();
/**
* write xml/yml formated parameters information
* @rparam fs : the filename of the xml file that will be open and writen with formatted parameters information
*/
virtual void write( String fs ) const;
/**
* write xml/yml formated parameters information
* @param fs : a cv::Filestorage object ready to be filled
*/
virtual void write( cv::FileStorage& fs ) const;
/**
* main processing method, get result using methods getSegmentationPicture()
* @param inputToSegment : the image to process, it must match the instance buffer size !
* @param channelIndex : the channel to process in case of multichannel images
*/
void run(InputArray inputToSegment, const int channelIndex=0);
/**
* access function
* @return the last segmentation result: a boolean picture which is resampled between 0 and 255 for a display purpose
*/
void getSegmentationPicture(OutputArray transientAreas);
/**
* cleans all the buffers of the instance
*/
void clearAllBuffers();
protected:
/**
* main processing method
* @param inputToSegment : the image to process as a valarray buffer it must match the instance buffer size !
* @param channelIndex : the channel to process in case of multichannel images
*/
void _run(const std::valarray<float> &inputToSegment, const int channelIndex=0);
/**
* access function
* @return the local motion energy level picture (experimental, not usefull)
*/
inline const std::valarray<float> &getLocalMotionPicture() const {return _localMotion;};
/**
* access function
* @return the neighborhood motion energy level picture (experimental, not usefull)
*/
inline const std::valarray<float> &getNeighborhoodMotionPicture() const {return _neighborhoodMotion;};
/**
* access function
* @return the motion energy context level picture (experimental, not usefull)
*/
inline const std::valarray<float> &getMotionContextPicture() const {return _contextMotionEnergy;};
struct cv::bioinspired::TransientAreasSegmentationModule::SegmentationParameters _segmentationParameters;
// template buffers and related acess pointers
std::valarray<float> _inputToSegment;
std::valarray<float> _contextMotionEnergy;
std::valarray<bool> _segmentedAreas;
// pointers to base class buffers
std::valarray<float> &_localMotion;
std::valarray<float> &_neighborhoodMotion;
unsigned int _numberOfSegmentedObjects;
cv::Mat _conversionBuffer;
cv::Mat _segmentedPicture;
// Buffer conversion utilities
void _convertValarrayBuffer2cvMat(const std::valarray<bool> &grayMatrixToConvert, const unsigned int nbRows, const unsigned int nbColumns, OutputArray outBuffer);
bool _convertCvMat2ValarrayBuffer(InputArray inputMat, std::valarray<float> &outputValarrayMatrix);
};
class TransientAreasSegmentationModuleImpl_: public TransientAreasSegmentationModule
{
public:
TransientAreasSegmentationModuleImpl_(const Size size):_segmTool(size){};
inline virtual Size getSize(){return _segmTool.getSize();};
inline virtual void write( cv::FileStorage& fs ) const{_segmTool.write(fs);};
inline virtual void setup(String segmentationParameterFile, const bool applyDefaultSetupOnFailure){_segmTool.setup(segmentationParameterFile, applyDefaultSetupOnFailure);};
inline virtual void setup(cv::FileStorage &fs, const bool applyDefaultSetupOnFailure){_segmTool.setup(fs, applyDefaultSetupOnFailure);};
inline virtual void setup(TransientAreasSegmentationModule::SegmentationParameters newParameters){_segmTool.setup(newParameters);};
inline virtual const String printSetup(){return _segmTool.printSetup();};
inline virtual struct TransientAreasSegmentationModule::TransientAreasSegmentationModule::SegmentationParameters getParameters(){return _segmTool.getParameters();};
inline virtual void write( String fs ) const{_segmTool.write(fs);};
inline virtual void run(InputArray inputToSegment, const int channelIndex){_segmTool.run(inputToSegment, channelIndex);};
inline virtual void getSegmentationPicture(OutputArray transientAreas){return _segmTool.getSegmentationPicture(transientAreas);};
inline virtual void clearAllBuffers(){_segmTool.clearAllBuffers();};
private:
TransientAreasSegmentationModuleImpl _segmTool;
};
/**
* allocator
* @param Size : size of the images input to segment (output will be the same size)
*/
Ptr<TransientAreasSegmentationModule> createTransientAreasSegmentationModule(Size inputSize){
return makePtr<TransientAreasSegmentationModuleImpl_>(inputSize);
};
// Constructor and destructors
TransientAreasSegmentationModuleImpl::TransientAreasSegmentationModuleImpl(const Size size)
:BasicRetinaFilter(size.height, size.width, 3),
// allocate the output of the class
_inputToSegment(size.height*size.width),
_contextMotionEnergy(size.height*size.width),
_segmentedAreas(size.height*size.width),
// set the pointer to the 2 frame buffer to the correct adress:
// -> the first low pass filter buffer will be _localBuffer
// -> the second will be _filterOutput;
_localMotion(_localBuffer),
_neighborhoodMotion(_filterOutput)
{
// default parameters setup
setup(_segmentationParameters);
//clean before running
clearAllBuffers();
}
TransientAreasSegmentationModuleImpl::~TransientAreasSegmentationModuleImpl()
{
}
void TransientAreasSegmentationModuleImpl::clearAllBuffers()
{
// flush parent buffers
bioinspired::BasicRetinaFilter::clearAllBuffers();
// flush instance buffers
_contextMotionEnergy=0;
_segmentedAreas=0;
}
struct TransientAreasSegmentationModule::SegmentationParameters TransientAreasSegmentationModuleImpl::getParameters()
{
return _segmentationParameters;
};
// setup from XML file
void TransientAreasSegmentationModuleImpl::setup(String segmentationParameterFile, const bool applyDefaultSetupOnFailure)
{
try
{
// opening retinaParameterFile in read mode
cv::FileStorage fs(segmentationParameterFile, cv::FileStorage::READ);
setup(fs, applyDefaultSetupOnFailure);
}catch(cv::Exception &e)
{
printf("Retina::setup: wrong/unappropriate xml parameter file : error report :`n=>%s\n", e.what());
if (applyDefaultSetupOnFailure)
{
printf("Retina::setup: resetting retina with default parameters\n");
cv::bioinspired::TransientAreasSegmentationModule::SegmentationParameters defaults;
setup(defaults);
}
else
{
printf("=> keeping current parameters");
}
}
}
// setup from cv::Filestorage object
void TransientAreasSegmentationModuleImpl::setup(cv::FileStorage &fs, const bool applyDefaultSetupOnFailure)
{
try
{
// read parameters file if it exists or apply default setup if asked for
if (!fs.isOpened())
{
std::cout<<"Retina::setup: provided parameters file could not be open... skeeping configuration"<<std::endl;
return;
// implicit else case : retinaParameterFile could be open (it exists at least)
}
// OPL and Parvo init first... update at the same time the parameters structure and the retina core
cv::FileNode rootFn = fs.root(), currFn=rootFn["SegmentationModuleSetup"];
currFn["thresholdON"]>>_segmentationParameters.thresholdON;
currFn["thresholdOFF"]>>_segmentationParameters.thresholdOFF;
currFn["localEnergy_temporalConstant"]>>_segmentationParameters.localEnergy_temporalConstant;
currFn["localEnergy_spatialConstant"]>>_segmentationParameters.localEnergy_spatialConstant;
currFn["neighborhoodEnergy_temporalConstant"]>>_segmentationParameters.neighborhoodEnergy_temporalConstant;
currFn["neighborhoodEnergy_spatialConstant"]>>_segmentationParameters.neighborhoodEnergy_spatialConstant;
currFn["contextEnergy_temporalConstant"]>>_segmentationParameters.contextEnergy_temporalConstant;
currFn["contextEnergy_spatialConstant"]>>_segmentationParameters.contextEnergy_spatialConstant;
setup(_segmentationParameters);
}catch(cv::Exception &e)
{
std::cout<<"Retina::setup: resetting retina with default parameters"<<std::endl;
if (applyDefaultSetupOnFailure)
{
struct cv::bioinspired::TransientAreasSegmentationModule::SegmentationParameters defaults;
setup(defaults);
}
std::cout<<"SegmentationModule::setup: wrong/unappropriate xml parameter file : error report :`n=>"<<e.what()<<std::endl;
std::cout<<"=> keeping current parameters"<<std::endl;
}
// report current configuration
printf("%s\n", printSetup().c_str());
}
// setup parameters for the 2 filters that allow the segmentation
void TransientAreasSegmentationModuleImpl::setup(cv::bioinspired::TransientAreasSegmentationModule::SegmentationParameters newParameters)
{
// copy structure contents
memcpy(&_segmentationParameters, &newParameters, sizeof(cv::bioinspired::TransientAreasSegmentationModule::SegmentationParameters));
// apply setup
// init local motion energy extraction low pass filter
BasicRetinaFilter::setLPfilterParameters(0, newParameters.localEnergy_temporalConstant, newParameters.localEnergy_spatialConstant);
// init neighbohood motion energy extraction low pass filter
BasicRetinaFilter::setLPfilterParameters(0, newParameters.neighborhoodEnergy_temporalConstant, newParameters.neighborhoodEnergy_spatialConstant, 1);
// init large area low pass filter
BasicRetinaFilter::setLPfilterParameters(0, newParameters.contextEnergy_temporalConstant, newParameters.contextEnergy_spatialConstant, 2);
}
const String TransientAreasSegmentationModuleImpl::printSetup()
{
std::stringstream outmessage;
outmessage<<"Current segmentation instance setup :"
<<"\n\t thresholdON : " << _segmentationParameters.thresholdON
<<"\n\t thresholdOFF : " << _segmentationParameters.thresholdOFF
<<"\n\t localEnergy_temporalConstant : " << _segmentationParameters.localEnergy_temporalConstant
<<"\n\t localEnergy_spatialConstant : " << _segmentationParameters.localEnergy_spatialConstant
<<"\n\t neighborhoodEnergy_temporalConstant : " << _segmentationParameters.neighborhoodEnergy_temporalConstant
<<"\n\t neighborhoodEnergy_spatialConstant : " << _segmentationParameters.neighborhoodEnergy_spatialConstant
<<"\n\t contextEnergy_temporalConstant : " << _segmentationParameters.contextEnergy_temporalConstant
<<"\n\t contextEnergy_spatialConstant : " << _segmentationParameters.contextEnergy_spatialConstant;
return outmessage.str().c_str();
}
void TransientAreasSegmentationModuleImpl::write( String fs ) const
{
cv::FileStorage parametersSaveFile(fs, cv::FileStorage::WRITE );
write(parametersSaveFile);
}
void TransientAreasSegmentationModuleImpl::write( cv::FileStorage& fs ) const
{
if (!fs.isOpened())
return; // basic error case
fs <<"SegmentationModuleSetup"<<"{";
fs <<"thresholdON" << _segmentationParameters.thresholdON;
fs <<"thresholdOFF" << _segmentationParameters.thresholdOFF;
fs <<"localEnergy_temporalConstant" << _segmentationParameters.localEnergy_temporalConstant;
fs <<"localEnergy_spatialConstant" << _segmentationParameters.localEnergy_spatialConstant;
fs <<"neighborhoodEnergy_temporalConstant" << _segmentationParameters.neighborhoodEnergy_temporalConstant;
fs <<"neighborhoodEnergy_spatialConstant" << _segmentationParameters.neighborhoodEnergy_spatialConstant;
fs <<"contextEnergy_temporalConstant" << _segmentationParameters.contextEnergy_temporalConstant;
fs <<"contextEnergy_spatialConstant" << _segmentationParameters.contextEnergy_spatialConstant;
fs <<"}";
}
void TransientAreasSegmentationModuleImpl::run(InputArray inputToProcess, const int channelIndex)
{
cv::Mat inputToSegment=inputToProcess.getMat();
// preliminary basic error check
if ( (inputToSegment.rows*inputToSegment.cols) != (int)_inputToSegment.size())
{
std::stringstream errorMsg;
errorMsg<<"Input matrix size does not match instance buffers setup !"
<<"\n\t Input size is : "<<inputToSegment.rows*inputToSegment.cols
<<"\n\t v.s. internalBuffer size is : "<< _inputToSegment.size();
throw cv::Exception(-1, errorMsg.str().c_str(), "SegmentationModule::run", "SegmentationModule.cpp", 0);
}
if (channelIndex >= inputToSegment.channels())
{
std::stringstream errorMsg;
errorMsg<<"Cannot access channel index "<<channelIndex<<" on the input matrix with channels quantity = "<<inputToSegment.channels();
throw cv::Exception(-1, errorMsg.str().c_str(), "SegmentationModule::run", "SegmentationModule.cpp", 0);
}
// create a cv::Mat header for the input valarray
// convert to float AND fill the valarray buffer
typedef float T; // define here the target pixel format, here, float
const int dsttype = cv::DataType<T>::depth; // output buffer is float format
cv::Mat dst(inputToSegment.size(), dsttype, &_inputToSegment[0]);
inputToSegment.convertTo(dst, dsttype);
//cv::imshow("Mask",dst);
//cv::waitKey();
// call the low level method
_run(_inputToSegment, channelIndex);
}
void TransientAreasSegmentationModuleImpl::_run(const std::valarray<float> &inputToSegment, const int channelIndex)
{
#ifdef SEGMENTATIONDEBUG
std::cout<<"Input length vs internal buffers length = "<<inputToSegment.size()<<", "<<_localMotion.size()<<std::endl;
#endif
// preliminary basic error check
// FIXME validate basic tests
//if (inputToSegment.size() != _localMotion.size())
// throw cv::Exception(-1, "Input matrix size does not match instance buffers setup !", "SegmentationModule::run", "SegmentationModule.cpp", 0);
// first square the input in order to increase the signal to noise ratio
// get motion local energy
_squaringSpatiotemporalLPfilter(&inputToSegment[channelIndex*getNBpixels()], &_localMotion[0]);
// second low pass filter: access to the neighborhood motion energy
_spatiotemporalLPfilter(&_localMotion[0], &_neighborhoodMotion[0], 1);
// third low pass filter: access to the background motion energy
_spatiotemporalLPfilter(&_localMotion[0], &_contextMotionEnergy[0], 2);
// compute the ON and OFF ways (positive and negative values of the difference of the two filterings)
float*localMotionPTR=&_localMotion[0], *neighborhoodMotionPTR=&_neighborhoodMotion[0], *contextMotionPTR=&_contextMotionEnergy[0];
// float meanEnergy=LPfilter2.sum()/(float)_LPfilter2.size();
register bool *segmentationPicturePTR= &_segmentedAreas[0];
for (unsigned int index=0; index<_filterOutput.getNBpixels() ; ++index, ++segmentationPicturePTR, ++localMotionPTR, ++neighborhoodMotionPTR, contextMotionPTR++)
{
float generalMotionContextDecision=*neighborhoodMotionPTR-*contextMotionPTR;
if (generalMotionContextDecision>0) // local maximum should be detected in this case
{
/* apply segmentation on local motion superior to its neighborhood
* => to segment objects moving faster than their neighborhood
*/
if (generalMotionContextDecision>_segmentationParameters.thresholdON)// && meanEnergy*1.1<*neighborhoodMotionPTR)
{
*segmentationPicturePTR=((*localMotionPTR-*neighborhoodMotionPTR)>_segmentationParameters.thresholdON);
}
else
*segmentationPicturePTR=false;
}
#ifdef USE_LOCALMINIMUMS
else // local minimum should be detected in this case
{
/* apply segmentation for non moving objects
* only if the wide area around motion energy is high
* => to segment object moving slower than the neighborhood
*/
if (-1.0*generalMotionContextDecision>_segmentationParameters.thresholdOFF && meanEnergy*0.9>*neighborhoodMotionPTR)
{
/* in order to segment non moving objects in a camera motion case
* we focus on local energy which is much lower than the wide neighborhood
*/
*segmentationPicturePTR+=(*neighborhoodMotionPTR-*localMotionPTR>_segmentationParameters.thresholdOFF)*127;
}
}
#else
else
*segmentationPicturePTR=false;
#endif
}
/*
#ifdef SEGMENTATIONDEBUG
std::cout<<"ON: max, min="<<_localMotionON.min()<<", "<<_localMotionON.max();
std::cout<<"/// \\\ OFF: max, min="<<_localMotionOFF.min()<<", "<<_localMotionOFF.max()<<std::endl;
std::cout<<"/// \\\ motion: max, min="<<_globalMotionEnergy.min()<<", "<<_globalMotionEnergy.max()<<std::endl;
std::cout<<"/// \\\ thresholds: ON, OFF="<<_thresholdON<<", "<<_thresholdOFF<<", meanEnergy= "<<meanEnergy<<std::endl;
#endif
*/
}
void TransientAreasSegmentationModuleImpl::getSegmentationPicture(OutputArray transientAreas)
{
_convertValarrayBuffer2cvMat(_segmentedAreas, getNBrows(), getNBcolumns(), transientAreas);
}
void TransientAreasSegmentationModuleImpl::_convertValarrayBuffer2cvMat(const std::valarray<bool> &grayMatrixToConvert, const unsigned int nbRows, const unsigned int nbColumns, OutputArray outBuffer)
{
// fill output buffer with the valarray buffer
const bool *valarrayPTR=get_data(grayMatrixToConvert);
outBuffer.create(cv::Size(nbColumns, nbRows), CV_8U);
Mat outMat = outBuffer.getMat();
for (unsigned int i=0;i<nbRows;++i)
{
for (unsigned int j=0;j<nbColumns;++j)
{
cv::Point2d pixel(j,i);
outMat.at<unsigned char>(pixel)=(unsigned char)*(valarrayPTR++);
}
}
}
bool TransientAreasSegmentationModuleImpl::_convertCvMat2ValarrayBuffer(InputArray inputMat, std::valarray<float> &outputValarrayMatrix)
{
const Mat inputMatToConvert=inputMat.getMat();
// first check input consistency
if (inputMatToConvert.empty())
throw cv::Exception(-1, "RetinaImpl cannot be applied, input buffer is empty", "RetinaImpl::run", "RetinaImpl.h", 0);
// retreive color mode from image input
int imageNumberOfChannels = inputMatToConvert.channels();
// convert to float AND fill the valarray buffer
typedef float T; // define here the target pixel format, here, float
const int dsttype = DataType<T>::depth; // output buffer is float format
const unsigned int nbPixels=inputMat.getMat().rows*inputMat.getMat().cols;
const unsigned int doubleNBpixels=inputMat.getMat().rows*inputMat.getMat().cols*2;
if(imageNumberOfChannels==4)
{
// create a cv::Mat table (for RGBA planes)
cv::Mat planes[4] =
{
cv::Mat(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[doubleNBpixels]),
cv::Mat(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[nbPixels]),
cv::Mat(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[0])
};
planes[3] = cv::Mat(inputMatToConvert.size(), dsttype); // last channel (alpha) does not point on the valarray (not usefull in our case)
// split color cv::Mat in 4 planes... it fills valarray directely
cv::split(Mat_<Vec<T, 4> >(inputMatToConvert), planes);
}
else if (imageNumberOfChannels==3)
{
// create a cv::Mat table (for RGB planes)
cv::Mat planes[] =
{
cv::Mat(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[doubleNBpixels]),
cv::Mat(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[nbPixels]),
cv::Mat(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[0])
};
// split color cv::Mat in 3 planes... it fills valarray directely
cv::split(cv::Mat_<Vec<T, 3> >(inputMatToConvert), planes);
}
else if(imageNumberOfChannels==1)
{
// create a cv::Mat header for the valarray
cv::Mat dst(inputMatToConvert.size(), dsttype, &outputValarrayMatrix[0]);
inputMatToConvert.convertTo(dst, dsttype);
}
else
CV_Error(Error::StsUnsupportedFormat, "input image must be single channel (gray levels), bgr format (color) or bgra (color with transparency which won't be considered");
return imageNumberOfChannels>1; // return bool : false for gray level image processing, true for color mode
}
}} //namespaces end : cv and bioinspired
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment