Seguimiento de objeto por su color

Author: devcarmelo

track-color/CMakeLists.txt

@@ -0,0 +1,5 @@
+cmake_minimum_required(VERSION 2.8)
+project( TrackColor )
+find_package( OpenCV REQUIRED )
+add_executable( TrackColor TrackColor.cpp )
+target_link_libraries( TrackColor ${OpenCV_LIBS} )

track-color/TrackColor.cpp

@@ -0,0 +1,71 @@
+#include <iostream>
+#include <string>
+#include <vector>
+
+#include <opencv2/highgui/highgui.hpp>
+#include <opencv2/imgproc/imgproc.hpp>
+
+using namespace cv;
+using namespace std;
+
+int main()
+{
+    Mat img, hsv, binary;
+
+    VideoCapture cap(0);
+
+    // verificar si se ha podido acceder a la web cam
+    if (!cap.isOpened()) {
+        cout << "Error al acceder a la web cam." << endl;
+        system("pause");
+        return -1;
+    }
+
+    while (true)
+    {
+        // obtener frame de la web cam
+        cap >> img;
+
+        // convertir imagen RGB a HSV 
+        cvtColor(img, hsv, CV_BGR2HSV);
+
+        // aplicar filtro para color deseado
+        inRange(hsv, Scalar(110, 50, 50), Scalar(130, 255, 255), binary);
+
+        // aplicar tranformaciones morfologicas (extrae la region de interes)
+        Mat element = getStructuringElement(MORPH_RECT, Size(15, 15));
+        erode(binary, binary, element);
+        dilate(binary, binary, element);
+
+        // buscar contornos en la imagen binaria
+        vector< vector<Point> > contours;
+        findContours(binary, contours, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_SIMPLE);        
+
+        // dibujar todos los contornos encontrados
+        drawContours(binary, contours, -1, Scalar(255), CV_FILLED);
+
+        // dibujar rectangulo y texto con coordenadas (x, y)
+        for (vector<Point> contour : contours) {
+            // dibujar rectangulo
+            Rect r = boundingRect(contour);
+            rectangle(img, r, CV_RGB(255, 0, 0), 2, CV_AA, 0);
+
+            Point center(r.x + (r.width / 2), r.y + (r.height / 2));
+
+            ostringstream str;
+            str << center.x << "," << center.y;
+
+            // mostrar texto (x, y)
+            putText(img, str.str(), center, FONT_HERSHEY_COMPLEX_SMALL, 0.60, CV_RGB(0, 255, 0), 1, CV_AA);
+        }
+
+        imshow("Track Color", img);
+        imshow("Track Color - Binary", binary);
+
+        // terminar la aplicacion con la tecla ESC
+        if(waitKey(30) == 27) break;
+      
+    }
+
+    return 0;
+}