miniprojet/opencv/detection.cpp

65 lines
1.4 KiB
C++

#include "opencv2/opencv.hpp"
using namespace cv;
using namespace std;
int main(int, char**)
{
int seuil=80;
char detect;
VideoCapture cap(0);
if(!cap.isOpened())
return -1;
namedWindow("Image",1);
namedWindow("Detection",1);
namedWindow("Contours",1);
for(;;) {
int X,Y,DIM,index,indexNB;
unsigned int numc;
uchar R,G,B;
vector<vector<Point> > contours;
vector<Vec4i> hierarchy;
Mat frame;
cap >> frame;
X=frame.rows;
Y=frame.cols;
Mat Binaire(X,Y,CV_8UC1);
imshow("Image", frame);
GaussianBlur(frame, frame, Size(7,7), 1.5, 1.5);
X=frame.rows;
Y=frame.cols;
DIM=frame.channels();
for (index=0,indexNB=0;index<DIM*X*Y;index+=DIM,indexNB++)
{
detect=0;
B=frame.data[index ];
G=frame.data[index + 1];
R=frame.data[index + 2];
if ((R>G) && (R>B))
if (((R-B)>=seuil) || ((R-G)>=seuil))
detect=1;
if (detect==1)
Binaire.data[indexNB]=255;
else
Binaire.data[indexNB]=0;
}
imshow("Detection", Binaire);
findContours(Binaire, contours, hierarchy, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE);
Mat Dessin = Mat::zeros(X,Y, CV_8UC1);
unsigned int max = 0;
int id = 0;
for(numc = 0; numc<contours.size(); numc++) {
if (contours[numc].size() > max) {
max = contours[numc].size();
id = numc;
}
}
drawContours(Dessin, contours, id, 255);
imshow("Contours", Dessin);
if(waitKey(30) == 27) {
break;
}
}
return 0;
}