jfmcarreira/calyp
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add motion based saliency and improve documentation

Browse Source
This commit is contained in:
jfmcarreira
2018-02-02 04:56:11 +00:00
parent 50cc22cb46
commit 2958c03249
4 changed files with 92 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/SaliencyDetection.cpp
+40 -4
View File
@@ -30,7 +30,7 @@
using cv::Mat;
using cv::Ptr;
using cv::saliency::Saliency;
using namespace cv::saliency;
SaliencyDetectionModule::SaliencyDetectionModule()
{
@@ -91,7 +91,7 @@ Bool SaliencyDetectionSpectral::create( std::vector<PlaYUVerFrame*> apcFrameList
if( !bRet )
return bRet;
m_ptrSaliencyAlgorithm = cv::saliency::StaticSaliencySpectralResidual::create();
m_ptrSaliencyAlgorithm = StaticSaliencySpectralResidual::create();
return true;
}
@@ -101,13 +101,18 @@ PlaYUVerFrame* SaliencyDetectionSpectral::process( std::vector<PlaYUVerFrame*> a
if( commonProcess( apcFrameList ) )
{
Mat matBinaryMap;
cv::saliency::StaticSaliencySpectralResidual spec;
StaticSaliencySpectralResidual spec;
spec.computeBinaryMap( m_matSaliency, matBinaryMap );
if( m_bBinaryMap )
{
m_pcSaliencyFrame->fromMat( matBinaryMap );
}
else
{
m_matSaliency *= 255;
m_matSaliency.convertTo( m_matSaliency, CV_8UC1 );
m_pcSaliencyFrame->fromMat( m_matSaliency );
}
}
return m_pcSaliencyFrame;
}
@@ -135,3 +140,34 @@ PlaYUVerFrame* SaliencyDetectionFineGrained::process( std::vector<PlaYUVerFrame*
m_pcSaliencyFrame->fromMat( m_matSaliency );
return m_pcSaliencyFrame;
}
SaliencyDetectionBinWangApr2014::SaliencyDetectionBinWangApr2014()
{
/* Module Definition */
m_pchModuleName = "SaliencyDetectionBinWangApr2014";
m_pchModuleLongName = "Fast Self-tuning Background Subtraction Algorithm";
m_pchModuleTooltip = "Measure saliency using a fast self-tuning background subtraction algorithm";
}
Bool SaliencyDetectionBinWangApr2014::create( std::vector<PlaYUVerFrame*> apcFrameList )
{
Bool bRet = commonCreate( apcFrameList );
if( !bRet )
return bRet;
m_ptrSaliencyAlgorithm = MotionSaliencyBinWangApr2014::create();
m_ptrSaliencyAlgorithm.dynamicCast<MotionSaliencyBinWangApr2014>()->setImagesize( apcFrameList[0]->getWidth(), apcFrameList[0]->getHeight() );
m_ptrSaliencyAlgorithm.dynamicCast<MotionSaliencyBinWangApr2014>()->init();
return true;
}
PlaYUVerFrame* SaliencyDetectionBinWangApr2014::process( std::vector<PlaYUVerFrame*> apcFrameList )
{
if( commonProcess( apcFrameList ) )
{
m_matSaliency *= 255;
m_matSaliency.convertTo( m_matSaliency, CV_8UC1 );
//cv::cvtColor(m_matSaliency, m_matSaliency, cv::COLOR_BGR2GRAY);
m_pcSaliencyFrame->fromMat( m_matSaliency );
}
return m_pcSaliencyFrame;
}