| #include "sequence_compensator_imp.h" |
|
|
| namespace Ivs |
| { |
| PtrSequenceCompensator CreateSequenceCompensator() |
| { |
| return std::make_unique<CSequenceCompensator>(); |
| } |
|
|
| CSequenceCompensator::~CSequenceCompensator() |
| { |
| m_matBackgroundMask.release(); |
| m_matBackground.release(); |
| } |
|
|
| void CSequenceCompensator::SetInputMeta(const TInputImageMeta& tInput) |
| { |
| m_matCurFrame = tInput.matImage; |
| m_vecRectObjRegion = tInput.vecObjRegions; |
|
|
| if (m_matBackground.empty()) |
| { |
| InitBackground(); |
| } |
| } |
|
|
| void CSequenceCompensator::Process() |
| { |
| if (m_matBackgroundMask.empty()) |
| { |
| return; |
| } |
|
|
| m_matCurFrame.copyTo(m_matBackground, m_matBackgroundMask); |
| UpdateBackgroundMask(); |
| } |
|
|
| bool CSequenceCompensator::IsBackgroundRecover() |
| { |
| if (m_matBackgroundMask.empty()) |
| { |
| return false; |
| } |
|
|
| return cv::countNonZero(m_matBackgroundMask) == 0; |
| } |
|
|
| cv::Mat CSequenceCompensator::GetBackgroundImage() |
| { |
| return m_matBackground; |
| } |
|
|
| void CSequenceCompensator::InitBackground() |
| { |
| m_matCurFrame.copyTo(m_matBackground); |
| m_matBackgroundMask = cv::Mat::zeros(m_matBackground.size(), CV_8UC1); |
|
|
| for (auto& rectObj : m_vecRectObjRegion) |
| { |
| m_matBackgroundMask(rectObj).setTo(255); |
| } |
| } |
|
|
| void CSequenceCompensator::UpdateBackgroundMask() |
| { |
|
|
| cv::Mat matMaskCur = cv::Mat::zeros(m_matBackgroundMask.size(), CV_8UC1); |
| for (auto& rect : m_vecRectObjRegion) |
| { |
| matMaskCur(rect).setTo(255); |
| } |
|
|
| cv::bitwise_and(m_matBackgroundMask, matMaskCur, m_matBackgroundMask); |
| m_vecRectObjRegionPrev = m_vecRectObjRegion; |
|
|
| matMaskCur.release(); |
| } |
|
|
| } |
|
|
