commons-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From t.@apache.org
Subject [math] Add userguide example for image clustering.
Date Mon, 28 Dec 2015 20:01:35 GMT
Repository: commons-math
Updated Branches:
  refs/heads/master 81585a3c4 -> f0943a724


Add userguide example for image clustering.


Project: http://git-wip-us.apache.org/repos/asf/commons-math/repo
Commit: http://git-wip-us.apache.org/repos/asf/commons-math/commit/f0943a72
Tree: http://git-wip-us.apache.org/repos/asf/commons-math/tree/f0943a72
Diff: http://git-wip-us.apache.org/repos/asf/commons-math/diff/f0943a72

Branch: refs/heads/master
Commit: f0943a7242041a16c8f7fca845592e48bb19ac74
Parents: 81585a3
Author: Thomas Neidhart <thomas.neidhart@gmail.com>
Authored: Mon Dec 28 21:01:18 2015 +0100
Committer: Thomas Neidhart <thomas.neidhart@gmail.com>
Committed: Mon Dec 28 21:01:18 2015 +0100

----------------------------------------------------------------------
 .../commons/math4/userguide/ExampleUtils.java   |  10 +
 .../clustering/ImageClusteringExample.java      | 203 +++++++++++++++++++
 src/userguide/resources/ColorfulBird.jpg        | Bin 0 -> 98227 bytes
 src/userguide/resources/references.txt          |   4 +-
 4 files changed, 216 insertions(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/commons-math/blob/f0943a72/src/userguide/java/org/apache/commons/math4/userguide/ExampleUtils.java
----------------------------------------------------------------------
diff --git a/src/userguide/java/org/apache/commons/math4/userguide/ExampleUtils.java b/src/userguide/java/org/apache/commons/math4/userguide/ExampleUtils.java
index e58540d..dddc96c 100644
--- a/src/userguide/java/org/apache/commons/math4/userguide/ExampleUtils.java
+++ b/src/userguide/java/org/apache/commons/math4/userguide/ExampleUtils.java
@@ -17,12 +17,14 @@
 package org.apache.commons.math4.userguide;
 
 import java.awt.Component;
+import java.awt.Graphics2D;
 import java.awt.event.ActionEvent;
 import java.awt.event.ActionListener;
 import java.awt.event.InputEvent;
 import java.awt.event.KeyEvent;
 import java.awt.image.BufferedImage;
 import java.io.File;
+import java.io.IOException;
 
 import javax.imageio.ImageIO;
 import javax.swing.JFileChooser;
@@ -100,5 +102,13 @@ public class ExampleUtils {
         component.paint(image.getGraphics());
         return image;
     }
+    
+    public static BufferedImage resizeImage(BufferedImage originalImage, int width, int height,
int type) throws IOException {
+        BufferedImage resizedImage = new BufferedImage(width, height, type);
+        Graphics2D g = resizedImage.createGraphics();
+        g.drawImage(originalImage, 0, 0, width, height, null);
+        g.dispose();
+        return resizedImage;
+    }
 
 }

http://git-wip-us.apache.org/repos/asf/commons-math/blob/f0943a72/src/userguide/java/org/apache/commons/math4/userguide/clustering/ImageClusteringExample.java
----------------------------------------------------------------------
diff --git a/src/userguide/java/org/apache/commons/math4/userguide/clustering/ImageClusteringExample.java
b/src/userguide/java/org/apache/commons/math4/userguide/clustering/ImageClusteringExample.java
new file mode 100644
index 0000000..67e197a
--- /dev/null
+++ b/src/userguide/java/org/apache/commons/math4/userguide/clustering/ImageClusteringExample.java
@@ -0,0 +1,203 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.commons.math4.userguide.clustering;
+
+import java.awt.BorderLayout;
+import java.awt.Color;
+import java.awt.Component;
+import java.awt.Dimension;
+import java.awt.FlowLayout;
+import java.awt.Graphics;
+import java.awt.GridLayout;
+import java.awt.event.ActionEvent;
+import java.awt.event.ActionListener;
+import java.awt.image.BufferedImage;
+import java.awt.image.Raster;
+import java.awt.image.WritableRaster;
+import java.io.File;
+import java.util.ArrayList;
+import java.util.List;
+
+import javax.imageio.ImageIO;
+import javax.swing.BorderFactory;
+import javax.swing.Box;
+import javax.swing.ImageIcon;
+import javax.swing.JButton;
+import javax.swing.JLabel;
+import javax.swing.JPanel;
+import javax.swing.JSpinner;
+import javax.swing.SpinnerNumberModel;
+
+import org.apache.commons.math4.ml.clustering.CentroidCluster;
+import org.apache.commons.math4.ml.clustering.Clusterable;
+import org.apache.commons.math4.ml.clustering.KMeansPlusPlusClusterer;
+import org.apache.commons.math4.userguide.ExampleUtils;
+import org.apache.commons.math4.userguide.ExampleUtils.ExampleFrame;
+
+/**
+ * This example shows how clustering can be applied to images.
+ */
+@SuppressWarnings("serial")
+public class ImageClusteringExample {
+
+    public static class Display extends ExampleFrame {
+        
+        private BufferedImage referenceImage;
+        private BufferedImage clusterImage;
+        
+        private Raster referenceRaster;
+        
+        private ImagePainter painter;
+        
+        private JSpinner clusterSizeSpinner;
+
+        public Display() throws Exception {
+            setTitle("Commons-Math: Image Clustering Example");
+            setSize(900, 350);
+            
+            setLayout(new FlowLayout());
+
+            Box bar = Box.createHorizontalBox();
+
+            referenceImage = ExampleUtils.resizeImage(
+                    ImageIO.read(new File("resources/ColorfulBird.jpg")),
+                    350,
+                    240,
+                    BufferedImage.TYPE_INT_RGB);
+
+            referenceRaster = referenceImage.getData();
+
+            clusterImage = new BufferedImage(referenceImage.getWidth(),
+                                             referenceImage.getHeight(),
+                                             BufferedImage.TYPE_INT_RGB);
+
+            JLabel picLabel = new JLabel(new ImageIcon(referenceImage));
+            bar.add(picLabel);
+
+            painter = new ImagePainter(clusterImage.getWidth(), clusterImage.getHeight());
+            bar.add(painter);
+
+            JPanel controlBox = new JPanel();
+            controlBox.setLayout(new GridLayout(5, 1));
+            controlBox.setBorder(BorderFactory.createLineBorder(Color.black, 1));
+            
+            JPanel sizeBox = new JPanel();
+            JLabel sizeLabel = new JLabel("Clusters:");
+            sizeBox.add(sizeLabel);
+            
+            SpinnerNumberModel model = new SpinnerNumberModel(3, 2, 10, 1);
+            clusterSizeSpinner = new JSpinner(model);
+            
+            sizeLabel.setLabelFor(clusterSizeSpinner);
+            sizeBox.add(clusterSizeSpinner);
+            controlBox.add(sizeBox, BorderLayout.NORTH);
+
+            JButton startButton = new JButton("Cluster");
+            startButton.setActionCommand("cluster");
+            controlBox.add(startButton, BorderLayout.CENTER);
+
+            bar.add(controlBox);
+
+            add(bar);
+
+            startButton.addActionListener(new ActionListener() {
+                public void actionPerformed(ActionEvent e) {
+                    clusterImage();
+                }
+            });
+        }
+        
+        private void clusterImage() {
+            List<PixelClusterable> pixels = new ArrayList<PixelClusterable>();
+            for (int row = 0; row < referenceImage.getHeight(); row++) {
+                for (int col = 0; col < referenceImage.getWidth(); col++) {
+                    pixels.add(new PixelClusterable(col, row));
+                }
+            }
+            
+            int clusterSize = ((Number) clusterSizeSpinner.getValue()).intValue();
+            KMeansPlusPlusClusterer<PixelClusterable> clusterer =
+                    new KMeansPlusPlusClusterer<PixelClusterable>(clusterSize);
+            List<CentroidCluster<PixelClusterable>> clusters = clusterer.cluster(pixels);
+            
+            WritableRaster raster = clusterImage.getRaster();
+            for (CentroidCluster<PixelClusterable> cluster : clusters) {
+                double[] color = cluster.getCenter().getPoint();
+                for (PixelClusterable pixel : cluster.getPoints()) {
+                    raster.setPixel(pixel.x, pixel.y, color);
+                }
+            }
+            
+            Display.this.repaint();
+        }
+
+        private class PixelClusterable implements Clusterable {
+            
+            private final int x;
+            private final int y;
+            private double[] color;
+            
+            public PixelClusterable(int x, int y) {
+                this.x = x;
+                this.y = y;
+                this.color = null;
+            }
+
+            @Override
+            public double[] getPoint() {
+                if (color == null) {
+                    color = referenceRaster.getPixel(x, y, (double[]) null);
+                }
+                return color;
+            }
+
+        }
+
+        private class ImagePainter extends Component {
+            
+            private int width;
+            private int height;
+
+            public ImagePainter(int width, int height) {
+                this.width = width;
+                this.height = height;
+            }
+
+            public Dimension getPreferredSize() {
+                return new Dimension(width, height);
+            }
+
+            @Override
+            public Dimension getMinimumSize() {
+                return getPreferredSize();
+            }
+
+            @Override
+            public Dimension getMaximumSize() {
+                return getPreferredSize();
+            }
+
+            public void paint(Graphics g) {
+                g.drawImage(clusterImage, 0, 0, this);
+            }
+        }
+    }
+
+    public static void main(String[] args) throws Exception {
+        ExampleUtils.showExampleFrame(new Display());
+    }
+}

http://git-wip-us.apache.org/repos/asf/commons-math/blob/f0943a72/src/userguide/resources/ColorfulBird.jpg
----------------------------------------------------------------------
diff --git a/src/userguide/resources/ColorfulBird.jpg b/src/userguide/resources/ColorfulBird.jpg
new file mode 100644
index 0000000..5753596
Binary files /dev/null and b/src/userguide/resources/ColorfulBird.jpg differ

http://git-wip-us.apache.org/repos/asf/commons-math/blob/f0943a72/src/userguide/resources/references.txt
----------------------------------------------------------------------
diff --git a/src/userguide/resources/references.txt b/src/userguide/resources/references.txt
index e61925f..f05294b 100644
--- a/src/userguide/resources/references.txt
+++ b/src/userguide/resources/references.txt
@@ -1,2 +1,4 @@
 monalisa.png - http://commons.wikimedia.org/wiki/File:Mona_Lisa.jpg
-the image is public domain in the United States, and those countries with a copyright term
of life of the author plus 100 years or less.
\ No newline at end of file
+the image is public domain in the United States, and those countries with a copyright term
of life of the author plus 100 years or less.
+
+ColorfulBird.jpg - http://www.wall321.com/Animals/Birds/colorful_birds_tropical_head_3888x2558_wallpaper_6566
\ No newline at end of file


Mime
View raw message