Fuzzy C-Means聚合算法在图像分割(segmentation)和图像视觉处理中常常被用到聚合算法之
一本文是完全基于JAVA语言实现Fuzzy C-Means聚合算法,并可以运用到图像处理中实现简
单的对象提取。
一:数学原理
在解释数学原理之前,请先看看这个链接算是热身吧
http://home.deib.polimi.it/matteucc/Clustering/tutorial_html/cmeans.html。
看不懂没关系。我的解释足够详细,小学毕业都可以学会,本人就是小学毕业。
Fuzzy C-means算法主要是比较RGB空间的每个像素值与Cluster中的每个中心点值,最终给
每个像素指派一个值(0~1之间)说明该像素更接近于哪里Cluster的中心点,模糊规则是该像
素对所有cluster的值之和为1。简单的举例:假设图像中有三个聚类cluster1,cluster2,cluster3,
像素A对三个聚类的值分别为a1, a2, a3, 根据模糊规则a1 + a2 + a3 = 1。更进一步,如果a1
最大,则该像素比较接近于Cluster1。计算总的对象值J:
二:算法流程
初始输入参数:
a. 指定的聚类个数numberOfClusters,
b. 指定的最大循环次数maxIteration
c. 指定的最小终止循环差值deltaValue
大致流程如下:
1. 初始化所有像素点值与随机选取每个Cluster的中心点,初始化每个像素点P[i]对应
Cluster的模糊值p[i][k]并计算cluster index。
2. 计算对象值J
3. 计算每个Cluster的颜色值,产生新的图像像素
4. 计算每个像素的对应每个cluster的模糊值,更新每个像素的Cluster Index
5. 再次计算对象值J,并与第二步的对象值相减,如果差值小于deltaValue或者达到最大
循环数,停止计算输出结果图像,否则继续2 ~ 4
三:关键代码解析
欧几里德距离计算方法如下:
private double calculateEuclideanDistance(ClusterPoint p, ClusterCentroid c)
{
// int pa = (p.getPixelColor() >> 24) & 0xff;
int pr = (p.getPixelColor() >> 16) & 0xff;
int pg = (p.getPixelColor() >> 8) & 0xff;
int pb = p.getPixelColor() & 0xff;
// int ca = (c.getPixelColor() >> 24) & 0xff;
int cr = (c.getPixelColor() >> 16) & 0xff;
int cg = (c.getPixelColor() >> 8) & 0xff;
int cb = c.getPixelColor() & 0xff;
return Math.sqrt(Math.pow((pr - cr), 2.0) + Math.pow((pg - cg), 2.0) + Math.pow((pb - cb), 2.0));
}
计算每个像素与每个Cluster的Fuzzy数值的代码如下:
public void stepFuzzy()
{
for (int c = 0; c < this.clusters.size(); c++)
{
for (int h = 0; h < this.points.size(); h++)
{
double top;
top = calculateEuclideanDistance(this.points.get(h), this.clusters.get(c));
if (top < 1.0) top = Eps;
// sumTerms is the sum of distances from this data point to all clusters.
double sumTerms = 0.0;
for (int ck = 0; ck < this.clusters.size(); ck++)
{
sumTerms += top / calculateEuclideanDistance(this.points.get(h), this.clusters.get(ck));
}
// Then the membership value can be calculated as...
fuzzyForPixels[h][c] = (double)(1.0 / Math.pow(sumTerms, (2 / (this.fuzzy - 1))));
}
};
this.recalculateClusterMembershipValues();
}
计算并更新每个像素的Cluster index的代码如下:
private void recalculateClusterMembershipValues()
{
for (int i = 0; i < this.points.size(); i++)
{
double max = 0.0;
double min = 0.0;
double sum = 0.0;
double newmax = 0;
ClusterPoint p = this.points.get(i);
//Normalize the entries
for (int j = 0; j < this.clusters.size(); j++)
{
max = fuzzyForPixels[i][j] > max ? fuzzyForPixels[i][j] : max;
min = fuzzyForPixels[i][j] < min ? fuzzyForPixels[i][j] : min;
}
//Sets the values to the normalized values between 0 and 1
for (int j = 0; j < this.clusters.size(); j++)
{
fuzzyForPixels[i][j] = (fuzzyForPixels[i][j] - min) / (max - min);
sum += fuzzyForPixels[i][j];
}
//Makes it so that the sum of all values is 1
for (int j = 0; j < this.clusters.size(); j++)
{
fuzzyForPixels[i][j] = fuzzyForPixels[i][j] / sum;
if (Double.isNaN(fuzzyForPixels[i][j]))
{
fuzzyForPixels[i][j] = 0.0;
}
newmax = fuzzyForPixels[i][j] > newmax ? fuzzyForPixels[i][j] : newmax;
}
// ClusterIndex is used to store the strongest membership value to a cluster, used for defuzzification
p.setClusterIndex(newmax);
};
}
四:运行效果
五:算法源代码
FuzzyCMeansProcessor - 算法类
package com.gloomyfish.segmentation.fuzzycmeans;
import java.awt.image.BufferedImage;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import com.gloomyfish.filter.study.AbstractBufferedImageOp;
public class FuzzyCMeansProcessor extends AbstractBufferedImageOp {
private List points;
private List clusters;
private BufferedImage originalImage;
private BufferedImage processedImage;
private double Eps = Math.pow(10, -5);
private double[][] fuzzyForPixels;
// Gets or sets objective function
private double numObj;
public void setObj(double j) {
this.numObj = j;
}
public double getObj() {
return this.numObj;
}
private float fuzzy; // default is 2
private int numCluster; // number of clusters in image
public BufferedImage getResultImage()
{
return this.processedImage;
}
public FuzzyCMeansProcessor(/*List points, List clusters, */float fuzzy, BufferedImage myImage, int numCluster)
{
points = new ArrayList();
int width = myImage.getWidth();
int height = myImage.getHeight();
int index = 0;
int[] inPixels = new int[width*height];
myImage.getRGB( 0, 0, width, height, inPixels, 0, width );
for (int row = 0; row < myImage.getHeight(); ++row)
{
for (int col = 0; col < myImage.getWidth(); ++col)
{
index = row * width + col;
int color = inPixels[index];
points.add(new ClusterPoint(row, col, color));
}
}
clusters = new ArrayList();
//Create random points to use a the cluster centroids
Random random = new Random();
for (int i = 0; i < numCluster; i++)
{
int randomNumber1 = random.nextInt(width);
int randomNumber2 = random.nextInt(height);
index = randomNumber2 * width + randomNumber1;
clusters.add(new ClusterCentroid(randomNumber1, randomNumber2, inPixels[index]));
}
this.originalImage = myImage;
this.fuzzy = fuzzy;
this.numCluster = numCluster;
double diff;
// Iterate through all points to create initial U matrix
fuzzyForPixels = new double[this.points.size()][this.clusters.size()];
for (int i = 0; i < this.points.size(); i++)
{
ClusterPoint p = points.get(i);
double sum = 0.0;
for (int j = 0; j < this.clusters.size(); j++)
{
ClusterCentroid c = this.clusters.get(j);
diff = Math.sqrt(Math.pow(calculateEuclideanDistance(p, c), 2.0));
fuzzyForPixels[i][j] = (diff == 0) ? Eps : diff;
sum += fuzzyForPixels[i][j];
}
}
// re-calculate the membership value for one point of all clusters, and make suer it's sum of value is 1
recalculateClusterMembershipValues();
}
public void calculateClusterCentroids()
{
for (int j = 0; j < this.clusters.size(); j++)
{
ClusterCentroid clusterCentroid = this.clusters.get(j);
double l = 0.0;
clusterCentroid.setRedSum(0);
clusterCentroid.setBlueSum(0);
clusterCentroid.setGreenSum(0);
clusterCentroid.setMemberShipSum(0);
double redSum = 0;
double greenSum = 0;
double blueSum = 0;
double memebershipSum = 0;
double pixelCount = 1;
for (int i = 0; i < this.points.size(); i++)
{
ClusterPoint p = this.points.get(i);
l = Math.pow(fuzzyForPixels[i][j], this.fuzzy);
int ta = (p.getPixelColor() >> 24) & 0xff;
int tr = (p.getPixelColor() >> 16) & 0xff;
int tg = (p.getPixelColor() >> 8) & 0xff;
int tb = p.getPixelColor() & 0xff;
redSum += l * tr;
greenSum += l * tg;
blueSum += l * tb;
memebershipSum += l;
if (fuzzyForPixels[i][j] == p.getClusterIndex())
{
pixelCount += 1;
}
}
int clusterColor = (255 << 24) | ((int)(redSum / memebershipSum) << 16) | ((int)(greenSum / memebershipSum) << 8) | (int)(blueSum / memebershipSum);
clusterCentroid.setPixelColor(clusterColor);
}
//update the original image
// Bitmap tempImage = new Bitmap(myImageWidth, myImageHeight, PixelFormat.Format32bppRgb);
BufferedImage tempImage = createCompatibleDestImage( originalImage, null );
int width = tempImage.getWidth();
int height = tempImage.getHeight();
int index = 0;
int[] outPixels = new int[width*height];
for (int j = 0; j < this.points.size(); j++)
{
for (int i = 0; i < this.clusters.size(); i++)
{
ClusterPoint p = this.points.get(j);
if (fuzzyForPixels[j][i] == p.getClusterIndex())
{
int row = (int)p.getX(); // row
int col = (int)p.getY(); // column
index = row * width + col;
outPixels[index] = this.clusters.get(i).getPixelColor();
}
}
}
// fill the pixel data
setRGB( tempImage, 0, 0, width, height, outPixels );
processedImage = tempImage;
}
///
/// Perform one step of the algorithm
///
public void stepFuzzy()
{
for (int c = 0; c < this.clusters.size(); c++)
{
for (int h = 0; h < this.points.size(); h++)
{
double top;
top = calculateEuclideanDistance(this.points.get(h), this.clusters.get(c));
if (top < 1.0) top = Eps;
// sumTerms is the sum of distances from this data point to all clusters.
double sumTerms = 0.0;
for (int ck = 0; ck < this.clusters.size(); ck++)
{
sumTerms += top / calculateEuclideanDistance(this.points.get(h), this.clusters.get(ck));
}
// Then the membership value can be calculated as...
fuzzyForPixels[h][c] = (double)(1.0 / Math.pow(sumTerms, (2 / (this.fuzzy - 1))));
}
};
this.recalculateClusterMembershipValues();
}
public double calculateObjectiveFunction()
{
double Jk = 0.0;
for (int i = 0; i < this.points.size();i++)
{
for (int j = 0; j < this.clusters.size(); j++)
{
Jk += Math.pow(fuzzyForPixels[i][j], this.fuzzy) * Math.pow(this.calculateEuclideanDistance(points.get(i), clusters.get(j)), 2);
}
}
return Jk;
}
private void recalculateClusterMembershipValues()
{
for (int i = 0; i < this.points.size(); i++)
{
double max = 0.0;
double min = 0.0;
double sum = 0.0;
double newmax = 0;
ClusterPoint p = this.points.get(i);
//Normalize the entries
for (int j = 0; j < this.clusters.size(); j++)
{
max = fuzzyForPixels[i][j] > max ? fuzzyForPixels[i][j] : max;
min = fuzzyForPixels[i][j] < min ? fuzzyForPixels[i][j] : min;
}
//Sets the values to the normalized values between 0 and 1
for (int j = 0; j < this.clusters.size(); j++)
{
fuzzyForPixels[i][j] = (fuzzyForPixels[i][j] - min) / (max - min);
sum += fuzzyForPixels[i][j];
}
//Makes it so that the sum of all values is 1
for (int j = 0; j < this.clusters.size(); j++)
{
fuzzyForPixels[i][j] = fuzzyForPixels[i][j] / sum;
if (Double.isNaN(fuzzyForPixels[i][j]))
{
fuzzyForPixels[i][j] = 0.0;
}
newmax = fuzzyForPixels[i][j] > newmax ? fuzzyForPixels[i][j] : newmax;
}
// ClusterIndex is used to store the strongest membership value to a cluster, used for defuzzification
p.setClusterIndex(newmax);
};
}
private double calculateEuclideanDistance(ClusterPoint p, ClusterCentroid c)
{
// int pa = (p.getPixelColor() >> 24) & 0xff;
int pr = (p.getPixelColor() >> 16) & 0xff;
int pg = (p.getPixelColor() >> 8) & 0xff;
int pb = p.getPixelColor() & 0xff;
// int ca = (c.getPixelColor() >> 24) & 0xff;
int cr = (c.getPixelColor() >> 16) & 0xff;
int cg = (c.getPixelColor() >> 8) & 0xff;
int cb = c.getPixelColor() & 0xff;
return Math.sqrt(Math.pow((pr - cr), 2.0) + Math.pow((pg - cg), 2.0) + Math.pow((pb - cb), 2.0));
}
@Override
public BufferedImage filter(BufferedImage src, BufferedImage dest) {
return processedImage;
}
}
ClusterPoint- 存储图像像素点对象
package com.gloomyfish.segmentation.fuzzycmeans;
public class ClusterPoint {
private double x;
private double y;
private int pixelColor;
private int originalPixelColor;
private double clusterIndex;
public ClusterPoint(double x, double y, int col)
{
this.x = x;
this.y = y;
this.pixelColor = col;
this.originalPixelColor = col;
this.clusterIndex = -1;
}
public double getX() {
return x;
}
public void setX(double x) {
this.x = x;
}
public double getY() {
return y;
}
public void setY(double y) {
this.y = y;
}
public int getPixelColor() {
return pixelColor;
}
public void setPixelColor(int pixelColor) {
this.pixelColor = pixelColor;
}
public int getOriginalPixelColor() {
return originalPixelColor;
}
public void setOriginalPixelColor(int originalPixelColor) {
this.originalPixelColor = originalPixelColor;
}
public double getClusterIndex() {
return clusterIndex;
}
public void setClusterIndex(double clusterIndex) {
this.clusterIndex = clusterIndex;
}
}
ClusterCentroid - 存储Cluster信息对象
package com.gloomyfish.segmentation.fuzzycmeans;
public class ClusterCentroid {
private double x;
private double y;
private int pixelColor;
private double redSum;
private double greenSum;
private double blueSum;
private double memberShipSum;
private int originalPixelColor;
public ClusterCentroid(double x, double y, int color)
{
this.x = x;
this.y = y;
this.originalPixelColor = color;
this.pixelColor = color;
}
public double getX() {
return x;
}
public void setX(double x) {
this.x = x;
}
public double getY() {
return y;
}
public void setY(double y) {
this.y = y;
}
public int getPixelColor() {
return pixelColor;
}
public void setPixelColor(int pixelColor) {
this.pixelColor = pixelColor;
}
public double getRedSum() {
return redSum;
}
public void setRedSum(double redSum) {
this.redSum = redSum;
}
public double getGreenSum() {
return greenSum;
}
public void setGreenSum(double greenSum) {
this.greenSum = greenSum;
}
public double getBlueSum() {
return blueSum;
}
public void setBlueSum(double blueSum) {
this.blueSum = blueSum;
}
public double getMemberShipSum() {
return memberShipSum;
}
public void setMemberShipSum(double memberShipSum) {
this.memberShipSum = memberShipSum;
}
public int getOriginalPixelColor() {
return originalPixelColor;
}
public void setOriginalPixelColor(int originalPixelColor) {
this.originalPixelColor = originalPixelColor;
}
}
算法调用:
int numClusters = 2; // (int)numericUpDown2.Value;
int maxIterations = 20; //(int)numericUpDown3.Value;
double accuracy = 0.00001; // (double)numericUpDown4.Value;
FuzzyCMeansProcessor alg = new FuzzyCMeansProcessor(numClusters, sourceImage, numClusters);
int k = 0;
do
{
k++;
alg.setObj(alg.calculateObjectiveFunction());
alg.calculateClusterCentroids();
alg.stepFuzzy();
double Jnew = alg.calculateObjectiveFunction();
System.out.println("Run method accuracy of delta value = " + Math.abs(alg.getObj() - Jnew));
if (Math.abs(alg.getObj() - Jnew) < accuracy) break;
}
while (maxIterations > k);
resultImage = alg.getResultImage();
this.repaint();
}
六:Fuzzy C-means不足之处
需要提供额外的参数,不能自动识别Cluster,运行时间比较长。
博客从本月开始更新,请关注!!谢谢谢!