Compare commits

..
Author SHA1 Message Date
Mariusz Pacek c6eb9dd872 Dodanie wyjasnien 2013-08-08 19:46:08 +02:00
Mariusz Pacek 08aaedecb5 poprawka 2013-06-14 12:19:12 +02:00
Mariusz Pacek cb0b231b4a poprawka 2013-06-14 11:57:02 +02:00
3 changed files with 107 additions and 174 deletions

No files matched your search

Binary file not shown.
+105 -173
View File
@@ -27,62 +27,18 @@
#include <stdlib.h>
#include<stdio.h>
#include <math.h>
#include <cvblob.h>
using namespace cv;
using namespace std;
using namespace cvb;
Mat frame, img1;
Point point;
Point matchLoc;
int drag = 0;
Point point; //punkt klikniecia mysza
Point matchLoc; //znaleźiony obszar przez matchtemplate
int drag = 0;
int key = 0;
CvRect rect;
bool done;
int searchSize = 20;
Point p1, p2, p3, p4;
Mat imgTemplate, searchRegion, rawTemplate, rawSearch, tran;
Rect solution;
bool noHand = false;
int r = 51, g = 99, b = 57;
int rT = 90, gT = 255, bT = 129;
int maxArea = 10000, minArea = 500;
bool hideWindow = false;
IplImage *red = 0, *tmp = 0, * framee = 0, *labelImg = 0;
CvBlobs blobs;
CvTracks tracks;
void setR(int v) {
r = v;
}
void setG(int v) {
g = v;
}
void setB(int v) {
b = v;
}
void setRT(int v) {
rT = v;
}
void setGT(int v) {
gT = v;
}
void setBT(int v) {
bT = v;
}
void setMax(int v) {
maxArea = v;
}
void setMin(int v) {
minArea = v;
}
CvRect rect; //prostokat po nacisnieciu myszka ?
bool done; //pierwszy etap skonczyony uzytkownik klikna myszka na zadany obszar na kltce
void mouseHandler(int event, int x, int y, int flags, void* param) {
if (event == CV_EVENT_LBUTTONUP) {
@@ -93,83 +49,26 @@ void mouseHandler(int event, int x, int y, int flags, void* param) {
void matchImage(Mat &img1, Mat &img2) {
Mat imgOutput;
int row = img1.rows - img2.rows + 1;
int row = img1.rows - img2.rows + 1; //wyliczenie ilosci pikseli w poziomie i w pionie
int col = img1.cols - img2.cols + 1;
imgOutput.create(col, row, CV_32FC1);
matchTemplate(img1, img2, imgOutput, CV_TM_CCOEFF_NORMED);
normalize(imgOutput, imgOutput, 0, 1, NORM_MINMAX, -1, Mat());
Point minLoc, maxLoc;
double minVal, maxVal;
minMaxLoc(imgOutput, &minVal, &maxVal, &minLoc, &maxLoc, Mat());
imgOutput.create(col, row, CV_32FC1); // stworzenie nowej kopii danego wycinka na ktorym bedzie szykany wzorzec
matchTemplate(img1, img2, imgOutput, CV_TM_CCOEFF_NORMED);
normalize(imgOutput, imgOutput, 0, 1, NORM_MINMAX, -1, Mat()); // to nie wiem po co jest
Point minLoc, maxLoc; // w zaleznosci od uzytej funkcji w matchTemplate zostanie uzyta pierwsz albo druga zmienna patrz dokumentacja w opencv
double minVal, maxVal;
minMaxLoc(imgOutput, &minVal, &maxVal, &minLoc, &maxLoc, Mat()); //wyznaczenie wspolrzednych na podstawie obrazu przychodzacego oraz wyliczonych danych
matchLoc = maxLoc;
}
void setRegion(Point point,int frameX,int frameY) {
int setRegionSize = 20;
if ((p1 = point.x - setRegionSize) > frameX ) p1 = frameX;
else
if ((p1 = point.x - setRegionSize) < 0) p1 = 0;
if ((p2 = point.y - setRegionSize) > frameY) p2 = frameY;
else
if ((p2 = point.y - setRegionSize) < 0) p2 = 0;
p1 = Point(point.x - 20, point.y - 20);
p2 = Point(point.x + 20, point.y + 20);
Rect ra = Rect(p1, p2);
imgTemplate = frame(ra);
p3 = Point(p1.x - 30, p1.y - 30);
p4 = Point(p2.x + 60, p2.y + 60);
searchRegion = frame(Rect(p3, p4));
rawTemplate = imgTemplate = imgTemplate.clone();
rawSearch = searchRegion = searchRegion.clone();
matchImage(searchRegion, imgTemplate);
solution = Rect(Point(matchLoc.x + p3.x, matchLoc.y + p3.y), Point(matchLoc.x + imgTemplate.cols + p3.x, matchLoc.y +
imgTemplate.rows + p3.y));
}
Point searchHand(Mat &img) {
cvtColor(frame, tran, CV_BGR2HSV);
tmp->imageData = (char*) tran.data;
cvInRangeS(tmp, Scalar(r, g, b), Scalar(rT, gT, bT), red);
cvSmooth(red, red, CV_BLUR, 3);
cvSmooth(red, red, CV_MEDIAN, 3);
cvDilate(red, red, NULL, 3);
cvErode(red, red, NULL, 8);
cvDilate(red, red, NULL, 2);
imshow("cos", Mat(red));
////cvtColor(frame, frame, CV_RGB2GRAY);
framee->imageData = (char*) img.data;
cvLabel(red, labelImg, blobs);
cvFilterByArea(blobs, minArea, maxArea);
cvRenderBlobs(labelImg, blobs, framee, framee, CV_BLOB_RENDER_BOUNDING_BOX);
cvUpdateTracks(blobs, tracks, 200, 5);
cvRenderTracks(tracks, framee, framee, CV_TRACK_RENDER_ID | CV_TRACK_RENDER_BOUNDING_BOX);
frame = Mat(framee);
cout << "jestem" << endl;
for (CvBlobs::const_iterator it = blobs.begin(); it != blobs.end(); ++it) {
int x = (int) it->second->centroid.x;
int y = (int) it->second->centroid.y;
return Point(x, y);
}
return Point(0, 0);
}
int main(int, char**) {
Media mouse;
bool mouseon = true;
int x, x1, y, y1;
int x, x1, y, y1; //punkty kotre beada tworzyc obwódkę ktora wskazuje na obszar prztwarzania
int region = 20; // to chyba niepotrzebne
int pointerSpeed = 3; //prszyspieszenie myszki x3
bool pointerOn = true,mouseOn = true; //sterowanie myszka, pokazywanie pointera on/off
// W H
pair<int, int> res = mouse.getResolution();
@@ -181,107 +80,134 @@ int main(int, char**) {
namedWindow("result", 1);
namedWindow("output", 1);
namedWindow("bin", 1);
cvCreateTrackbar("R", "bin", &r, 255, setR);
cvCreateTrackbar("G", "bin", &g, 255, setG);
cvCreateTrackbar("B", "bin", &b, 255, setB);
cvCreateTrackbar("RT", "bin", &rT, 255, setRT);
cvCreateTrackbar("BT", "bin", &bT, 255, setBT);
cvCreateTrackbar("GT", "bin", &gT, 255, setGT);
cvCreateTrackbar("min", "bin", &minArea, 1000, setMin);
cvCreateTrackbar("max", "bin", &maxArea, 100000, setMax);
cap >> frame;
int frameX = frame.cols, frameY = frame.rows;
int frameX = frame.cols, frameY = frame.rows; //rozdzielczosc kamerki
flip(frame, frame, 1);
red = cvCreateImage(cvSize(frame.cols, frame.rows), IPL_DEPTH_8U, 1);
tmp = cvCreateImage(cvSize(frame.cols, frame.rows), IPL_DEPTH_8U, 3);
labelImg = cvCreateImage(cvGetSize(red), IPL_DEPTH_LABEL, 1);
framee = cvCreateImage(cvGetSize(red), 8, 3);
Mat rawTemplate, rawSearch;
while (!done) {
cap >> frame;
flip(frame, frame, 1);
flip(frame, frame, 1); //odbicie lustrzane klatki (patrzac na monitor naturalnie sie porusza reka)
imshow("result", frame);
if (rect.width > 0)
frame = frame(rect);
setMouseCallback("result", mouseHandler, NULL);
frame = frame(rect); // tego nie wiem musisz sobie sprawdzic
setMouseCallback("result", mouseHandler, NULL); //przchwycenie eventu z myszki
imshow("result", frame);
cvWaitKey(10);
}
setRegion(point);
Point p1 = Point(point.x - 10, point.y - 10); //wyznaczenie wzorca o boku 20px
Point p2 = Point(point.x + 10, point.y + 10);
Rect ra = Rect(p1, p2);
Mat imgTemplate = frame(ra);
Point p3 = Point(p1.x - 30, p1.y - 30);//wyznaczenie obszaru do obrobki czyli bedzie to kwadrat gdzie w srodku znajduje startowy tamplate
Point p4 = Point(p2.x + 60, p2.y + 60);
Mat searchRegion = frame(Rect(p3, p4)); //wycinek obrazu na ktorym szukamy
rawTemplate = imgTemplate = imgTemplate.clone(); //kopia wzorca
rawSearch = searchRegion = searchRegion.clone(); //kopia obszaru
matchImage(searchRegion, imgTemplate); // szukaj wzorca
char key = -1;
Rect solution = Rect(Point(matchLoc.x + p3.x, matchLoc.y + p3.y), Point(matchLoc.x + imgTemplate.cols + p3.x, matchLoc.y +
imgTemplate.rows + p3.y)); //wylicz pozycje znaleźionego wzorca na podstawie duzego wycinka
/*
(0,0) frame.x
---------------------------
| p3(p1-30) |
| |----------| |f
| | p1 | |r
| | |--| | |a
| | |--|p2 | |m
| |--------- |p4(p2+60)|e
| |.
| |y
| |
| |
|-------------------------|
*/
while (true) {
cap >> frame;
flip(frame, frame, 1);
Point p1 = Point(matchLoc.x, matchLoc.y);
Point p2 = Point(matchLoc.x + imgTemplate.cols, matchLoc.y + imgTemplate.rows);
p1.x = matchLoc.x; //wspolzedne templata
p1.y = matchLoc.y;
p2.x = matchLoc.x + imgTemplate.cols; // druga para pobieram szerokosc i wysoksc tmaplata poniewaz chce miec taki sam obszar
p2.y = matchLoc.y + imgTemplate.rows;
imgTemplate = searchRegion.clone();//kopuje caly wycinek z obszaru poszukiwac
imgTemplate = imgTemplate(Rect(p1, p2)); // przycinam go do zadanych rozmiarow trzeba tak zrobic aby nie stracic obrzu z searchRegion
imgTemplate = searchRegion.clone();
imgTemplate = imgTemplate(Rect(p1, p2));
searchRegion = frame.clone();
searchRegion = frame.clone(); //analogicznie jak wyzej
searchRegion = searchRegion(Rect(p3, p4));
matchImage(searchRegion, imgTemplate); // poszukuj nowego wzorca
matchImage(searchRegion, imgTemplate);
solution = Rect(Point(matchLoc.x + p3.x, matchLoc.y + p3.y), Point(matchLoc.x + imgTemplate.cols + p3.x, matchLoc.y +
imgTemplate.rows + p3.y));
imgTemplate.rows + p3.y)); //jak wyzej
//solution.Rect_(p1.x + p3.x, p1.y + p3.x, p2.x + p3.x, p2.y + p3.y);
if ((x = solution.x - searchSize) <= 0)
/*
* sprawdzanie czy rectangle ktory sie rysuje na obrazie nie wychodzi po zaniego efekt widoczny podczas dzilania programu
* wodac jak niebieski kwadrat plywa przy krawedziach kaltaki oraz czerwnoy kwadrat sie automatycznie zmniejsza jezeli jest blizej krawedzi
* co powoduje ze program nie lata po niewiadomej pamieci i nie sypier errorami :D
*/
x = solution.x - region;
y = solution.y - region;
x1 = solution.x + solution.width + region;
y1 = solution.y + solution.height + region;
if (x <= 0)
x = 0;
else
if ((x = solution.x - searchSize) >= frameX)
if (x >= frameX)
x = frameX;
if ((y = solution.y - searchSize) <= 0)
if (y <= 0)
y = 0;
else
if ((y = solution.y - searchSize) >= frameY)
if (y >= frameY)
y = frameY;
if ((x1 = solution.x + solution.width + searchSize) <= 0)
if (x1 <= 0)
x1 = 0;
else
if ((x1 = solution.x + solution.width + searchSize) >= frameX)
if (x1 >= frameX)
x1 = frameX;
if ((y1 = solution.y + solution.height + searchSize) <= 0)
if (y1 <= 0)
y1 = 0;
else
if ((y1 = solution.y + solution.height + searchSize) >= frameY)
if (y1 >= frameY)
y1 = frameY;
p3 = Point(x, y);
p4 = Point(x1, y1);
if ((solution.x + solution.width == frameX || solution.y + solution.height ==
frameY || solution.x == 0 || solution.y == 0)) {
Point center = searchHand(frame);
setRegion(center);
}
rectangle(frame, p3, p4, Scalar(0, 0, 255), 2, 8, 0);
p3.x = x; //przypisanie do starych wartosci w taki sposob jest wygodniej bo dziala a podajac sama wspolrzedne mialem problem z wykmineniem kolejnosci wiec mi sie niechcialo tego robic no i jest czytelniej
p3.y = y;
p4.x = x1;
p4.y = y1;
rectangle(frame, p3, p4, Scalar(0, 0, 255), 2, 8, 0); //rysowanie na frame , scalar to jest wybor koloru , nastepne argumenty to jakies tam sytle i grubosci lini
rectangle(frame, solution, Scalar(255, 0, 0), 2, 8, 0);
x = (int) (solution.x + solution.x + solution.width) / 2;
x = (int) (solution.x + solution.x + solution.width) / 2; //srodka znalexionego wzorca
y = (int) (solution.y + solution.y + solution.height) / 2;
//mouse.setMouse(x / (double) frame.cols * res.first, y / (double) frame.rows * res.second, 3);
//mouse.drawArc(x / (double) frame.cols * res.first, y / (double) frame.rows * res.second, 3);
if(mouseOn)
mouse.setMouse(x / (double) frame.cols * res.first, y / (double) frame.rows * res.second, pointerSpeed);
if(pointerOn)
mouse.drawArc(x / (double) frame.cols * res.first, y / (double) frame.rows * res.second, pointerSpeed);
imshow("result3", searchRegion);
imshow("result3", searchRegion); //prezetacja wynikow
imshow("result", frame);
imshow("output", imgTemplate);
imshow("bin", Mat(red));
key = waitKey(1);
@@ -292,6 +218,12 @@ int main(int, char**) {
imgTemplate = rawTemplate.clone();
searchRegion = frame.clone();
break;
case 'p':
mouseOn = mouseOn ? false : true;
break;
case 'm':
pointerOn = pointerOn ? false : true;
break;
}
}
+2 -1
View File
@@ -101,7 +101,8 @@ void Media::drawArc(int _x, int _y, double speed) {
}
void Media::draw() {
XFillRectangle(dpy, root_window, gc, currXR, currYR, 50, 50);
}