c++ code

--------

main file

---------

#include <opencv2/core/core.hpp>

#include <opencv2/highgui/highgui.hpp>

#include "colordetector.h"

int main()

{

        // Create image processor object

        ColorDetector cdetect;

        // Read input image

        cv::Mat image= cv::imread("coloredchips.png");

        if (!image.data)

                return 0;

   // set input parameters

        cdetect.setTargetColor(23,23,242); // start with yellow

   // Read image, process it and display the result

        cv::namedWindow("result");

        cv::namedWindow("original");

        cv::imshow("result",cdetect.process(image));

        cv::imshow("original", image);

        cv::waitKey();

        return 0;

}

colordetector.h

---------------

#if !defined COLORDETECT

#define COLORDETECT

#include <opencv2/core/core.hpp>

class ColorDetector {

  private:

          // minimum acceptable distance

          int minDist;

          // target color

          cv::Vec3b target;

          // image containing resulting binary map

          cv::Mat result;

          // inline private member function

          // Computes the distance from target color.

          int getDistance(const cv::Vec3b& color) const {

                 // return static_cast<int>(cv::norm<int,3>(cv::Vec3i(color[0]-target[0],color[1]-$

                  return abs(color[0]-target[0])+

                                        abs(color[1]-target[1])+

                                        abs(color[2]-target[2]);

          }

  public:

          // empty constructor

          ColorDetector() : minDist(100) {

                  // default parameter initialization here

                  target[0]= target[1]= target[2]= 0;

          }

          // Getters and setters

          // Sets the color distance threshold.

          // Threshold must be positive, otherwise distance threshold

          // is set to 0.

          void setColorDistanceThreshold(int distance) {

                  if (distance<0)

                          distance=0;

                  minDist= distance;

          }

          // Gets the color distance threshold

          int getColorDistanceThreshold() const {

                  return minDist;

          }

         // Sets the color to be detected

          void setTargetColor(unsigned char red, unsigned char green, unsigned char blue) {

                  target[2]= red;

                  target[1]= green;

                  target[0]= blue;

          }

          // Sets the color to be detected

          void setTargetColor(cv::Vec3b color) {

                  target= color;

          }

          // Gets the color to be detected

          cv::Vec3b getTargetColor() const {

                  return target;

          }

          // Processes the image. Returns a 1-channel binary image.

          cv::Mat process(const cv::Mat &image);

};

#endif

colordetector.cpp

-----------------------

#include "colordetector.h"

cv::Mat ColorDetector::process(const cv::Mat &image) {

          // re-allocate binary map if necessary

          // same size as input image, but 1-channel

          result.create(image.rows,image.cols,CV_8U);

          // get the iterators

          cv::Mat_<cv::Vec3b>::const_iterator it= image.begin<cv::Vec3b>();

          cv::Mat_<cv::Vec3b>::const_iterator itend= image.end<cv::Vec3b>();

          cv::Mat_<uchar>::iterator itout= result.begin<uchar>();

          // for each pixel

          for ( ; it!= itend; ++it, ++itout) {

                // process each pixel ---------------------

                  // compute distance from target color

                  if (getDistance(*it)<minDist) {

                          *itout= 255;

                  } else {

                          *itout= 0;

                  }

        // end of pixel processing ----------------

          }

          return result;

}