动态人物识别代码

int main()
{// 打开视频文件或摄像头VideoCapture cap("vtest.avi");// VideoCapture cap(0);  // 使用默认摄像头if (!cap.isOpened()){std::cout << "无法打开视频文件或摄像头流" << std::endl;return -1;}// 读取第一帧Mat prevFrame,prevGray;cap >> prevFrame;prevGray = Mat(prevFrame.size(), CV_8UC1, Scalar(255));int kk = 0;while (true){Mat nextFrame, nextGray;cap >> nextFrame;if (nextFrame.empty())break;// 将当前帧与背景帧进行差分Mat diff;absdiff(prevFrame, nextFrame, diff);// 将差分图像转换为灰度图像Mat gray;cvtColor(diff, gray, COLOR_BGR2GRAY);GaussianBlur(gray, gray, Size(9, 9), 2, 2);  //平滑滤波// 对灰度图像进行阈值处理threshold(gray, nextGray, 30, 255, THRESH_BINARY);//当前二值化图像  --  （上一张和当前图像的二值化图像） 与 （下一张和当前图像的二值化图像）Mat currentGray;bitwise_and(prevGray, nextGray, currentGray);// 对二值图像进行膨胀操作，以填充物体区域Mat dilated;Mat kernel = getStructuringElement(MORPH_RECT, Size(3, 3));dilate(currentGray, dilated, kernel, Point(-1, -1), 3);//生成随机颜色，用于区分不同连通域RNG rng(10086);Mat out, stats, centroids;//统计图像中连通域的个数int number = connectedComponentsWithStats(dilated, out, stats, centroids, 8, CV_16U);vector<Vec3b> colors;for (int i = 0; i < number; i++){//使用均匀分布的随机数确定颜色Vec3b vec3 = Vec3b(rng.uniform(0, 256), rng.uniform(0, 256), rng.uniform(0, 256));colors.push_back(vec3);}//以不同颜色标记出不同的连通域Mat result = Mat::zeros(prevFrame.size(), CV_8UC3);int w = result.cols;int h = result.rows;for (int i = 1; i < number; i++){// 中心位置int center_x = centroids.at<double>(i, 0);int center_y = centroids.at<double>(i, 1);//矩形边框int x = stats.at<int>(i, CC_STAT_LEFT);int y = stats.at<int>(i, CC_STAT_TOP);int w = stats.at<int>(i, CC_STAT_WIDTH);int h = stats.at<int>(i, CC_STAT_HEIGHT);int area = stats.at<int>(i, CC_STAT_AREA);if (area < 500){continue;}// 外接矩形Rect rect(x, y, w, h);rectangle(prevFrame, rect, colors[i],2);}// 寻找轮廓//std::vector<std::vector<Point>> contours;//std::vector<Vec4i> hierarchy;//findContours(dilated, contours, hierarchy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);// 绘制检测到的轮廓//for (size_t i = 0; i < contours.size(); i++)//{//	if (minAreaRect(contours[i]).boundingRect().area()< 500)  // 设置最小轮廓面积以过滤噪声//		continue;//	Rect boundingRect1 = boundingRect(contours[i]);//	rectangle(prevFrame, boundingRect1, Scalar(0, 255, 0), 2);//}// 显示结果帧imshow("Motion Detection", prevFrame);// 按下ESC键退出循环if (waitKey(1) == 27)break;//用作下一个循环做对比prevGray = nextGray;prevFrame = nextFrame;}// 释放资源并关闭窗口cap.release();destroyAllWindows();return 0;
}