关闭

图像处理之K-Means算法演示

标签: 算法机器学习可视化
87人阅读 评论(0) 收藏 举报
分类:

一:数学原理

K-Means算法的作者是MacQueen, 基本的数学原理很容易理解,假设有一个像素

数据集P。我们要根据值不同将它分为两个基本的数据集合Cluster1, Cluster2,使

用K-Means算法大致如下:

假设两个Cluster的RGB值分别为112,225,244和23,34,99则像素集合中的像素点

a(222,212,234), b(198,205,229), c(25,77,52),d(34,55,101)计算每个像素点与这

两个cluster中心点的欧几里德距离,则像素点a, b属于前面一个cluster, c,d属于

后面一个cluster。然后在根据(222+198)/2, (212+205)/2, (234+52)/2更新cluster

的RGB值,对后一个cluster做同样处理。然后再计算每个像素点到cluster中心点

的欧几里德距离。最终值没有变化则得到分类Cluster点集合。

二:算法基本流程


三:算法关键代码解析

初始化cluster中心点代码如下:

[java] view plain copy
  1. Random random = new Random();  
  2. for (int i = 0; i < numOfCluster; i++)  
  3. {  
  4.     int randomNumber1 = random.nextInt(width);  
  5.     int randomNumber2 = random.nextInt(height);  
  6.     index = randomNumber2 * width + randomNumber1;  
  7.     ClusterCenter cc = new ClusterCenter(randomNumber1, randomNumber2, inPixels[index]);  
  8.     cc.setcIndex(i);  
  9.     clusterCenterList.add(cc);   
  10. }  

初始化所有像素点代码如下:

[java] view plain copy
  1. // create all cluster point  
  2. for (int row = 0; row < height; ++row)  
  3. {  
  4.     for (int col = 0; col < width; ++col)  
  5.     {  
  6.         index = row * width + col;  
  7.         int color = inPixels[index];  
  8.         pointList.add(new ClusterPoint(row, col, color));  
  9.   
  10.     }  
  11. }  

计算两个像素点之间欧几里德距离的代码如下:

[java] view plain copy
  1. // int pa = (p.getPixelColor() >> 24) & 0xff;  
  2. int pr = (p.getPixelColor() >> 16) & 0xff;  
  3. int pg = (p.getPixelColor() >> 8) & 0xff;  
  4. int pb = p.getPixelColor() & 0xff;  
  5. // int ca = (c.getPixelColor() >> 24) & 0xff;  
  6. int cr = (c.getPixelColor() >> 16) & 0xff;  
  7. int cg = (c.getPixelColor() >> 8) & 0xff;  
  8. int cb = c.getPixelColor() & 0xff;  
  9.   
  10. return Math.sqrt(Math.pow((pr - cr), 2.0) + Math.pow((pg - cg), 2.0) + Math.pow((pb - cb), 2.0));  
重新计算Cluster中心点RGB值的代码如下:

[java] view plain copy
  1. private double[] reCalculateClusterCenters() {  
  2.       
  3.     // clear the points now  
  4.     for(int i=0; i<clusterCenterList.size(); i++)  
  5.     {  
  6.          clusterCenterList.get(i).setNumOfPoints(0);  
  7.     }  
  8.       
  9.     // recalculate the sum and total of points for each cluster  
  10.     double[] redSums = new double[3];  
  11.     double[] greenSum = new double[3];  
  12.     double[] blueSum = new double[3];  
  13.     for(int i=0; i<pointList.size(); i++)  
  14.     {  
  15.         int cIndex = (int)pointList.get(i).getClusterIndex();  
  16.         clusterCenterList.get(cIndex).addPoints();  
  17.         int ta = (pointList.get(i).getPixelColor() >> 24) & 0xff;  
  18.         int tr = (pointList.get(i).getPixelColor() >> 16) & 0xff;  
  19.         int tg = (pointList.get(i).getPixelColor() >> 8) & 0xff;  
  20.         int tb = pointList.get(i).getPixelColor() & 0xff;  
  21.         ta = 255;  
  22.         redSums[cIndex] += tr;  
  23.         greenSum[cIndex] += tg;  
  24.         blueSum[cIndex] += tb;  
  25.     }  
  26.       
  27.     double[] oldClusterCentersColors = new double[clusterCenterList.size()];  
  28.     for(int i=0; i<clusterCenterList.size(); i++)  
  29.     {  
  30.         double sum  = clusterCenterList.get(i).getNumOfPoints();  
  31.         int cIndex = clusterCenterList.get(i).getcIndex();  
  32.         int red = (int)(greenSum[cIndex]/sum);  
  33.         int green = (int)(greenSum[cIndex]/sum);  
  34.         int blue = (int)(blueSum[cIndex]/sum);  
  35.         System.out.println("red = " + red + " green = " + green + " blue = " + blue);  
  36.         int clusterColor = (255 << 24) | (red << 16) | (green << 8) | blue;  
  37.         clusterCenterList.get(i).setPixelColor(clusterColor);  
  38.         oldClusterCentersColors[i] = clusterColor;  
  39.     }  
  40.       
  41.     return oldClusterCentersColors;  
  42. }  

四:运行效果


五:K-Means算法源代码

[java] view plain copy
  1. package com.gloomyfish.segmentation.kmeans;  
  2.   
  3. import java.awt.image.BufferedImage;  
  4. import java.util.ArrayList;  
  5. import java.util.List;  
  6. import java.util.Random;  
  7.   
  8. import com.gloomyfish.filter.study.AbstractBufferedImageOp;  
  9. import com.gloomyfish.segmentation.fuzzycmeans.ClusterPoint;  
  10.   
  11. public class KMeansProcessor extends AbstractBufferedImageOp {  
  12.     private List<ClusterCenter> clusterCenterList;  
  13.     private List<ClusterPoint> pointList;  
  14.       
  15.     private int numOfCluster;  
  16.       
  17.     public KMeansProcessor(int clusters)  
  18.     {  
  19.         this.numOfCluster = clusters;  
  20.         pointList = new ArrayList<ClusterPoint>();  
  21.         this.clusterCenterList = new ArrayList<ClusterCenter>();  
  22.     }  
  23.   
  24.     @Override  
  25.     public BufferedImage filter(BufferedImage src, BufferedImage dest) {  
  26.         // initialization the pixel data  
  27.         int width = src.getWidth();  
  28.         int height = src.getHeight();  
  29.         int[] inPixels = new int[width*height];  
  30.         src.getRGB( 00, width, height, inPixels, 0, width );  
  31.         int index = 0;  
  32.           
  33.         //Create random points to use a the cluster center  
  34.         Random random = new Random();  
  35.         for (int i = 0; i < numOfCluster; i++)  
  36.         {  
  37.             int randomNumber1 = random.nextInt(width);  
  38.             int randomNumber2 = random.nextInt(height);  
  39.             index = randomNumber2 * width + randomNumber1;  
  40.             ClusterCenter cc = new ClusterCenter(randomNumber1, randomNumber2, inPixels[index]);  
  41.             cc.setcIndex(i);  
  42.             clusterCenterList.add(cc);   
  43.         }  
  44.           
  45.         // create all cluster point  
  46.         for (int row = 0; row < height; ++row)  
  47.         {  
  48.             for (int col = 0; col < width; ++col)  
  49.             {  
  50.                 index = row * width + col;  
  51.                 int color = inPixels[index];  
  52.                 pointList.add(new ClusterPoint(row, col, color));  
  53.   
  54.             }  
  55.         }  
  56.           
  57.         // initialize the clusters for each point  
  58.         double[] clusterDisValues = new double[clusterCenterList.size()];  
  59.         for(int i=0; i<pointList.size(); i++)  
  60.         {  
  61.             for(int j=0; j<clusterCenterList.size(); j++)  
  62.             {  
  63.                 clusterDisValues[j] = calculateEuclideanDistance(pointList.get(i), clusterCenterList.get(j));  
  64.             }  
  65.             pointList.get(i).setClusterIndex(getCloserCluster(clusterDisValues));  
  66.         }  
  67.           
  68.         // calculate the old summary  
  69.         // assign the points to cluster center  
  70.         // calculate the new cluster center  
  71.         // computation the delta value  
  72.         // stop condition--  
  73.         double[] oldClusterCenterColors = reCalculateClusterCenters();  
  74.         while(true)  
  75.         {  
  76.             stepClusters();  
  77.             double[] newClusterCenterColors = reCalculateClusterCenters();  
  78.             if(isStop(oldClusterCenterColors, newClusterCenterColors))  
  79.             {                 
  80.                 break;  
  81.             }   
  82.             else  
  83.             {  
  84.                 oldClusterCenterColors = newClusterCenterColors;  
  85.             }  
  86.         }  
  87.           
  88.         //update the result image  
  89.         dest = createCompatibleDestImage(src, null );  
  90.         index = 0;  
  91.         int[] outPixels = new int[width*height];         
  92.         for (int j = 0; j < pointList.size(); j++)  
  93.         {  
  94.             for (int i = 0; i < clusterCenterList.size(); i++)  
  95.             {  
  96.                 ClusterPoint p = this.pointList.get(j);  
  97.                 if (clusterCenterList.get(i).getcIndex() == p.getClusterIndex())  
  98.                 {  
  99.                     int row = (int)p.getX(); // row  
  100.                     int col = (int)p.getY(); // column  
  101.                     index = row * width + col;  
  102.                     outPixels[index] = clusterCenterList.get(i).getPixelColor();  
  103.                 }  
  104.             }  
  105.         }  
  106.           
  107.         // fill the pixel data  
  108.         setRGB( dest, 00, width, height, outPixels );  
  109.         return dest;  
  110.     }  
  111.       
  112.     private boolean isStop(double[] oldClusterCenterColors, double[] newClusterCenterColors) {  
  113.         for(int i=0; i<oldClusterCenterColors.length; i++)  
  114.         {  
  115.             System.out.println("cluster " + i + " old : " + oldClusterCenterColors[i] + ", new : " + newClusterCenterColors[i]);  
  116.             if(oldClusterCenterColors[i]  != newClusterCenterColors[i])   
  117.             {  
  118.                 return false;  
  119.             }  
  120.         }  
  121.         System.out.println();  
  122.         return true;  
  123.     }  
  124.   
  125.     /** 
  126.      * update the cluster index by distance value 
  127.      */  
  128.     private void stepClusters()   
  129.     {  
  130.         // initialize the clusters for each point  
  131.         double[] clusterDisValues = new double[clusterCenterList.size()];  
  132.         for(int i=0; i<pointList.size(); i++)  
  133.         {  
  134.             for(int j=0; j<clusterCenterList.size(); j++)  
  135.             {  
  136.                 clusterDisValues[j] = calculateEuclideanDistance(pointList.get(i), clusterCenterList.get(j));  
  137.             }  
  138.             pointList.get(i).setClusterIndex(getCloserCluster(clusterDisValues));  
  139.         }  
  140.           
  141.     }  
  142.   
  143.     /** 
  144.      * using cluster color of each point to update cluster center color 
  145.      *  
  146.      * @return 
  147.      */  
  148.     private double[] reCalculateClusterCenters() {  
  149.           
  150.         // clear the points now  
  151.         for(int i=0; i<clusterCenterList.size(); i++)  
  152.         {  
  153.              clusterCenterList.get(i).setNumOfPoints(0);  
  154.         }  
  155.           
  156.         // recalculate the sum and total of points for each cluster  
  157.         double[] redSums = new double[3];  
  158.         double[] greenSum = new double[3];  
  159.         double[] blueSum = new double[3];  
  160.         for(int i=0; i<pointList.size(); i++)  
  161.         {  
  162.             int cIndex = (int)pointList.get(i).getClusterIndex();  
  163.             clusterCenterList.get(cIndex).addPoints();  
  164.             int ta = (pointList.get(i).getPixelColor() >> 24) & 0xff;  
  165.             int tr = (pointList.get(i).getPixelColor() >> 16) & 0xff;  
  166.             int tg = (pointList.get(i).getPixelColor() >> 8) & 0xff;  
  167.             int tb = pointList.get(i).getPixelColor() & 0xff;  
  168.             ta = 255;  
  169.             redSums[cIndex] += tr;  
  170.             greenSum[cIndex] += tg;  
  171.             blueSum[cIndex] += tb;  
  172.         }  
  173.           
  174.         double[] oldClusterCentersColors = new double[clusterCenterList.size()];  
  175.         for(int i=0; i<clusterCenterList.size(); i++)  
  176.         {  
  177.             double sum  = clusterCenterList.get(i).getNumOfPoints();  
  178.             int cIndex = clusterCenterList.get(i).getcIndex();  
  179.             int red = (int)(greenSum[cIndex]/sum);  
  180.             int green = (int)(greenSum[cIndex]/sum);  
  181.             int blue = (int)(blueSum[cIndex]/sum);  
  182.             System.out.println("red = " + red + " green = " + green + " blue = " + blue);  
  183.             int clusterColor = (255 << 24) | (red << 16) | (green << 8) | blue;  
  184.             clusterCenterList.get(i).setPixelColor(clusterColor);  
  185.             oldClusterCentersColors[i] = clusterColor;  
  186.         }  
  187.           
  188.         return oldClusterCentersColors;  
  189.     }  
  190.       
  191.       
  192.   
  193.     /** 
  194.      *  
  195.      * @param clusterDisValues 
  196.      * @return 
  197.      */  
  198.     private double getCloserCluster(double[] clusterDisValues)  
  199.     {  
  200.         double min = clusterDisValues[0];  
  201.         int clusterIndex = 0;  
  202.         for(int i=0; i<clusterDisValues.length; i++)  
  203.         {  
  204.             if(min > clusterDisValues[i])  
  205.             {  
  206.                 min = clusterDisValues[i];  
  207.                 clusterIndex = i;  
  208.             }  
  209.         }  
  210.         return clusterIndex;  
  211.     }  
  212.   
  213.     /** 
  214.      *  
  215.      * @param point 
  216.      * @param cluster 
  217.      * @return distance value 
  218.      */  
  219.     private double calculateEuclideanDistance(ClusterPoint p, ClusterCenter c)   
  220.     {  
  221.         // int pa = (p.getPixelColor() >> 24) & 0xff;  
  222.         int pr = (p.getPixelColor() >> 16) & 0xff;  
  223.         int pg = (p.getPixelColor() >> 8) & 0xff;  
  224.         int pb = p.getPixelColor() & 0xff;  
  225.         // int ca = (c.getPixelColor() >> 24) & 0xff;  
  226.         int cr = (c.getPixelColor() >> 16) & 0xff;  
  227.         int cg = (c.getPixelColor() >> 8) & 0xff;  
  228.         int cb = c.getPixelColor() & 0xff;  
  229.           
  230.         return Math.sqrt(Math.pow((pr - cr), 2.0) + Math.pow((pg - cg), 2.0) + Math.pow((pb - cb), 2.0));  
  231.     }  
  232.   
  233. }  
转载请注明出自gloomyfish博客
0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:26096次
    • 积分:576
    • 等级:
    • 排名:千里之外
    • 原创:29篇
    • 转载:15篇
    • 译文:0篇
    • 评论:9条
    最新评论