git.maemo.org Git - opencv/blob - samples/c/kmeans.c

   1 #ifdef _CH_
   2 #pragma package <opencv>
   3 #endif
   4
   5 #define CV_NO_BACKWARD_COMPATIBILITY
   6
   7 #ifndef _EiC
   8 #include "cv.h"
   9 #include "highgui.h"
  10 #include <stdio.h>
  11 #endif
  12
  13 int main( int argc, char** argv )
  14 {
  15     #define MAX_CLUSTERS 5
  16     CvScalar color_tab[MAX_CLUSTERS];
  17     IplImage* img = cvCreateImage( cvSize( 500, 500 ), 8, 3 );
  18     CvRNG rng = cvRNG(-1);
  19     CvPoint ipt;
  20
  21     color_tab[0] = CV_RGB(255,0,0);
  22     color_tab[1] = CV_RGB(0,255,0);
  23     color_tab[2] = CV_RGB(100,100,255);
  24     color_tab[3] = CV_RGB(255,0,255);
  25     color_tab[4] = CV_RGB(255,255,0);
  26
  27     cvNamedWindow( "clusters", 1 );
  28
  29     for(;;)
  30     {
  31         char key;
  32         int k, cluster_count = cvRandInt(&rng)%MAX_CLUSTERS + 1;
  33         int i, sample_count = cvRandInt(&rng)%1000 + 1;
  34         CvMat* points = cvCreateMat( sample_count, 1, CV_32FC2 );
  35         CvMat* clusters = cvCreateMat( sample_count, 1, CV_32SC1 );
  36         cluster_count = MIN(cluster_count, sample_count);
  37
  38         /* generate random sample from multigaussian distribution */
  39         for( k = 0; k < cluster_count; k++ )
  40         {
  41             CvPoint center;
  42             CvMat point_chunk;
  43             center.x = cvRandInt(&rng)%img->width;
  44             center.y = cvRandInt(&rng)%img->height;
  45             cvGetRows( points, &point_chunk, k*sample_count/cluster_count,
  46                        k == cluster_count - 1 ? sample_count :
  47                        (k+1)*sample_count/cluster_count, 1 );
  48
  49             cvRandArr( &rng, &point_chunk, CV_RAND_NORMAL,
  50                        cvScalar(center.x,center.y,0,0),
  51                        cvScalar(img->width*0.1,img->height*0.1,0,0));
  52         }
  53
  54         /* shuffle samples */
  55         for( i = 0; i < sample_count/2; i++ )
  56         {
  57             CvPoint2D32f* pt1 = (CvPoint2D32f*)points->data.fl + cvRandInt(&rng)%sample_count;
  58             CvPoint2D32f* pt2 = (CvPoint2D32f*)points->data.fl + cvRandInt(&rng)%sample_count;
  59             CvPoint2D32f temp;
  60             CV_SWAP( *pt1, *pt2, temp );
  61         }
  62
  63         printf( "iterations=%d\n", cvKMeans2( points, cluster_count, clusters,
  64                 cvTermCriteria( CV_TERMCRIT_EPS+CV_TERMCRIT_ITER, 10, 1.0 ),
  65                 5, 0, 0, 0, 0 ));
  66
  67         cvZero( img );
  68
  69         for( i = 0; i < sample_count; i++ )
  70         {
  71             int cluster_idx = clusters->data.i[i];
  72             ipt.x = (int)points->data.fl[i*2];
  73             ipt.y = (int)points->data.fl[i*2+1];
  74             cvCircle( img, ipt, 2, color_tab[cluster_idx], CV_FILLED, CV_AA, 0 );
  75         }
  76
  77         cvReleaseMat( &points );
  78         cvReleaseMat( &clusters );
  79
  80         cvShowImage( "clusters", img );
  81
  82         key = (char) cvWaitKey(0);
  83         if( key == 27 || key == 'q' || key == 'Q' ) // 'ESC'
  84             break;
  85     }
  86
  87     cvDestroyWindow( "clusters" );
  88     return 0;
  89 }
  90
  91 #ifdef _EiC
  92 main(1,"kmeans.c");
  93 #endif