gpt4 book ai didi

c++ - std::vector 每次使用时都填充相同的数据

转载 作者:太空宇宙 更新时间:2023-11-03 22:05:28 25 4
gpt4 key购买 nike

好的,所以下面的程序应该获取一个视频文件并将其分割成单独的帧,并将每个帧保存在它自己的文件中,但是不应保存所有帧,而应该只保存其中包含人员的帧,包括大约某人进入画面前 20 帧。

我的代码:

#include <opencv2/opencv.hpp>
#include <vector>

void checkForPerson(const cv::Mat &frame, bool &personInside, bool &personInsideLastFrame);
void saveFrame(const cv::Mat &frame, std::vector<const cv::Mat> &frameBuffer, const bool personInsideLastFrame, int &savedFramesCount);

int main(){
cv::VideoCapture vc;
cv::Mat currFrame, currFrameBackup, filtered, bg;
std::vector<const cv::Mat> frameBuffer;
const cv::String org = "Original";
const cv::String filt = "Filtered";
bool personInside = false;
bool personInsideLastFrame;
int savedFramesCount = 0;

vc.open("res/video1.avi");
if(!vc.isOpened()){
std::cerr << "cant load video file!\n";
system("pause");
return EXIT_FAILURE;
}

vc >> bg; // use first frame as background asuming no people are inside
if(!bg.data){
std::cerr << "no frames found\n";
system("pause");
return EXIT_FAILURE;
}
cv:cvtColor(bg, bg, cv::COLOR_BGR2GRAY);

int c;
while(vc.read(currFrame) && ((c = cv::waitKey(5)) != 27)){
cv::cvtColor(currFrame, currFrameBackup, cv::COLOR_BGR2GRAY);
filtered = bg - currFrameBackup;
cv::threshold(filtered, filtered, 0, 255, CV_THRESH_BINARY | CV_THRESH_OTSU);
cv::Mat erode = cv::getStructuringElement(cv::MORPH_RECT, cv::Size(3,3)); //just to clean up the mess a bit
cv::Mat dilate = cv::getStructuringElement(cv::MORPH_RECT, cv::Size(3,3));
cv::erode(filtered, filtered, erode);
cv::dilate(filtered, filtered, dilate);
checkForPerson(filtered, personInside, personInsideLastFrame);
if(personInside || personInsideLastFrame){
saveFrame(currFrame, frameBuffer, personInsideLastFrame, savedFramesCount);
}else{
if(frameBuffer.size() == 20){
frameBuffer.erase(frameBuffer.begin());
}
frameBuffer.push_back(currFrame);
}

cv::imshow(org, currFrame);
cv::imshow(filt, filtered);
}

return EXIT_SUCCESS;
}

void checkForPerson(const cv::Mat &frame, bool &personInside, bool &personInsideLastFrame){
if(personInside){
personInsideLastFrame = true;
}else{
personInsideLastFrame = false;
}
cv::MatConstIterator_<uchar> it = frame.begin<uchar>();
cv::MatConstIterator_<uchar> it_end = frame.end<uchar>();
for(; it != it_end; ++it){
if((*it) == 255){
personInside = true;
return;
}
}
personInside = false;
}

void saveFrame(const cv::Mat &frame, std::vector<const cv::Mat> &frameBuffer, const bool personInsideLastFrame, int &savedFramesCount){
if(personInsideLastFrame){
std::stringstream ss;
ss << "result/" << ++savedFramesCount << ".jpg";
cv::imwrite(ss.str(), frame);
}else{
for(int i = 0; i<frameBuffer.size(); ++i){
std::stringstream ss;
ss << "result/" << ++savedFramesCount << ".jpg";
cv::imwrite(ss.str(), frameBuffer[i]);
//cv::imshow("tmp", frameBuffer[i]);
//cv::waitKey(0);
}
frameBuffer.clear();
std::stringstream ss;
ss << "result/" << ++savedFramesCount << ".jpg";
cv::imwrite(ss.str(), frame);
}
}

代码中的 frameBuffer 用于 20(如果没有 20,则应使用最大可用)预输入帧以及补偿一些“丢失的帧”和里面的人。但是,当我将帧保存到第一帧时,似乎帧已正确插入到 frameBuffer 中,当检测到内部人员时保存了 20 次。当在 currentFrame 中没有检测到任何内容时,我将一个帧插入到 frameBuffer 中,并在最后一帧中没有检测到任何内容时使用这些帧。这真的炸了我的脑袋,我似乎看不到问题(实际上什么都看不到),所以有人请告诉我我犯了一个愚蠢的错误,我也会非常感激一些建议。

很抱歉,如果我的描述有些令人困惑,我找不到更好的解释方式。

最佳答案

除非需要,否则 OpenCV 不会重新分配 Mat。 Mat 本质上是指向缓冲区的智能指针。如果将 Mat 推回 vector ,这就像复制指向同一缓冲区的指针。然后你重新使用缓冲区, vector 中的指针指向新的内容!

将 Mat 推到 vector 上时需要克隆它:

frameBuffer.push_back(currFrame.clone());

关于c++ - std::vector 每次使用时都填充相同的数据,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25218670/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com