34 lines
749 B
C++
34 lines
749 B
C++
#include "ImageApplyRefuseInflow.h"
|
|
|
|
CImageApplyRefuseInflow::CImageApplyRefuseInflow(int d, int sigmaColor, int sigmaSpace)
|
|
: m_d(d)
|
|
, m_sigmaColor(sigmaColor)
|
|
, m_sigmaSpace(sigmaSpace)
|
|
{
|
|
}
|
|
|
|
CImageApplyRefuseInflow::~CImageApplyRefuseInflow()
|
|
{
|
|
}
|
|
|
|
void CImageApplyRefuseInflow::apply(cv::Mat& pDib, int side)
|
|
{
|
|
cv::Mat dst;
|
|
cv::bilateralFilter(pDib, dst, m_d, m_sigmaColor, m_sigmaSpace);
|
|
pDib = dst;
|
|
}
|
|
|
|
void CImageApplyRefuseInflow::apply(std::vector<cv::Mat>& mats, bool isTwoSide)
|
|
{
|
|
(void)isTwoSide;
|
|
|
|
int i = 0;
|
|
for (cv::Mat& var : mats) {
|
|
if (i != 0 && isTwoSide == false)
|
|
break;
|
|
if (!var.empty())
|
|
apply(var, 0);
|
|
i++;
|
|
}
|
|
}
|