0

我有这段代码，基本上在两帧上做了一个“哑”的背景减法。

void FrameDifferenceBGS::operator()(cv::InputArray _image, cv::OutputArray _fgmask, double learningRate)
{
  cv::Mat img_input = _image.getMat();

  if(img_input.empty())
    return;

  _fgmask.create(img_input.size(), CV_8U);
  cv::Mat img_foreground = _fgmask.getMat();

  if(img_input_prev.empty())
  {
    img_input.copyTo(img_input_prev);
    return;
  }

  cv::absdiff(img_input_prev, img_input, img_foreground);

  if(img_foreground.channels() == 3)
    cv::cvtColor(img_foreground, img_foreground, CV_BGR2GRAY);

  if(enableThreshold)
    cv::threshold(img_foreground, img_foreground, threshold, 255, cv::THRESH_BINARY);

  if(showOutput)
    cv::imshow("Frame Difference", img_foreground);

  img_input.copyTo(img_input_prev);
  img_foreground.copyTo(_fgmask);
  firstTime = false;
}

如果我最后不添加img_foreground.copyTo(_fgmask)，则输出数组不会使用 的结果进行更新img_foreground，从而在调用它时产生黑色图像。

我在做什么错/应该在这里做什么？

4

2 回答 2

1

使固定

  • 更改 _fgmask.create(img_input.size(), CV_8U);为 _fgmask.create(img_input.size(), CV_8UC3);或 _fgmask.create(img_input.size(), img_input.type());

为什么

  • 这是因为cv::absdiff(img_input_prev, img_input, img_foreground);每次在内部重新创建一个新数组。它确实更新了 img_foreground 结构，但在分配之后，_fgmask 内的内存地址数据无法更改，因为标头是按值传递的。
    • 你可以通过做看似解决这个问题（但仍然会产生创建成本）cv::Mat& img_foreground = _fgmask.getMatRef();
  • 那是因为 CV_8U 与 CV_8UC3 不同，因此 mat.hpp 中的检查 @ Mat::create() 总是由于类型差异而最终分配一个新数组

观点

我想......也许用垫子代替？

#include "opencv2/opencv.hpp"
using namespace cv;

class FrameDifferenceBGS
{
public:
    Mat prev;
    Mat diff;
    bool enableThreshold;
    bool showOutput;
    bool firstTime;
    uchar threshold;
    FrameDifferenceBGS():firstTime(false),enableThreshold(false),showOutput(false),threshold(0)
    {

    }
    void FrameDifferenceBGS::operator()(cv::Mat& _in, cv::Mat &_fg, double _lr)
    {
        if(_in.empty())
            return;

        if(prev.empty())
        {
            prev=_in.clone();
            _fg=cv::Mat::zeros(_in.size(),CV_8UC1);
            return;
        }

        cv::absdiff(prev, _in, diff);

        if(diff.channels() == 3)
            cv::cvtColor(diff, _fg, CV_BGR2GRAY);
        else
            _fg=diff;

        if(enableThreshold)
            cv::threshold(_fg, _fg, threshold, 255, cv::THRESH_BINARY);

        if(showOutput)
            cv::imshow("Frame Difference", _fg);

        prev=_in.clone();
        firstTime = false;
    }
};

int main()
{
    VideoCapture cap(0);
    FrameDifferenceBGS bgs;
    Mat frame,fg;
    for(;;)
    {
        cap >> frame; 
        bgs(frame,fg,0);

        imshow("frame", frame);
        imshow("fg", fg);
        if(waitKey(1) ==27) exit(0);
    }
    return 0;
}

编辑2（修改原件）

#include "opencv2/opencv.hpp"

class FrameDifferenceBGS
{
public:
    cv::Mat img_input_prev;
    cv::Mat diff;

    cv::Mat img_foreground;//put this in class in stead of inside the function
    bool enableThreshold;
    bool showOutput;
    bool firstTime;
    uchar threshold;
    FrameDifferenceBGS():firstTime(false),enableThreshold(false),showOutput(false),threshold(0)
    {

    }
    void FrameDifferenceBGS::operator()(cv::InputArray _image, cv::OutputArray _fgmask, double learningRate)
    {
        cv::Mat img_input = _image.getMat();

        if(img_input.empty())
            return;
        if(_fgmask.empty())
            _fgmask.create(img_input.size(), CV_8UC1);
        if(img_input_prev.empty())
        {
            img_input.copyTo(img_input_prev);
            return;
        }

        cv::absdiff(img_input_prev, img_input, img_foreground);

        if(img_foreground.channels() == 3)
            cv::cvtColor(img_foreground, _fgmask, CV_BGR2GRAY);

        if(enableThreshold)
            cv::threshold(img_foreground, img_foreground, threshold, 255, cv::THRESH_BINARY);

        if(showOutput)
            cv::imshow("Frame Difference", img_foreground);

        img_input.copyTo(img_input_prev);
        //img_foreground.copyTo(_fgmask);
        firstTime = false;
    }
};

int main()
{
    cv::VideoCapture cap(0);
    FrameDifferenceBGS bgs;
    cv::Mat frame,fg;
    for(;;)
    {
        cap >> frame; 
        bgs(frame,fg,0);

        cv::imshow("frame", frame);
        cv::imshow("fg", fg);
        if(cv::waitKey(1) ==27) exit(0);
    }
    return 0;
}
于 2013-10-02T17:15:51.247 回答
1

我再次查看了您的代码。看起来您正在为 _fgmask 创建新对象。

 _fgmask.create(img_input.size(), CV_8U);

我认为这就是您遇到问题的原因。因为这个论点中的引用不同于这个陈述之后的那个。为什么不在调用函数之前调用该行。

于 2013-10-02T15:50:36.967 回答