View Javadoc

1   /**
2    * Copyright (c) 2011, The University of Southampton and the individual contributors.
3    * All rights reserved.
4    *
5    * Redistribution and use in source and binary forms, with or without modification,
6    * are permitted provided that the following conditions are met:
7    *
8    *   * 	Redistributions of source code must retain the above copyright notice,
9    * 	this list of conditions and the following disclaimer.
10   *
11   *   *	Redistributions in binary form must reproduce the above copyright notice,
12   * 	this list of conditions and the following disclaimer in the documentation
13   * 	and/or other materials provided with the distribution.
14   *
15   *   *	Neither the name of the University of Southampton nor the names of its
16   * 	contributors may be used to endorse or promote products derived from this
17   * 	software without specific prior written permission.
18   *
19   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
20   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21   * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
22   * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
23   * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
24   * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25   * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
26   * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27   * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28   * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29   */
30  package org.openimaj.workinprogress.featlearn;
31  
32  import java.io.IOException;
33  import java.util.List;
34  
35  import org.openimaj.image.DisplayUtilities;
36  import org.openimaj.image.MBFImage;
37  import org.openimaj.image.annotation.evaluation.datasets.CIFAR10Dataset;
38  import org.openimaj.image.colour.RGBColour;
39  import org.openimaj.math.matrix.algorithm.whitening.ZCAWhitening;
40  import org.openimaj.math.statistics.normalisation.PerExampleMeanCenter;
41  import org.openimaj.ml.clustering.kmeans.SphericalKMeans;
42  import org.openimaj.ml.clustering.kmeans.SphericalKMeansResult;
43  
44  public class Test2 {
45  	public static void main(String[] args) throws IOException {
46  		System.out.println("start");
47  		final RandomPatchSampler<MBFImage> sampler = new RandomPatchSampler<MBFImage>(
48  				CIFAR10Dataset.getTrainingImages(CIFAR10Dataset.MBFIMAGE_READER),
49  				8, 8, 400000);
50  		final List<MBFImage> patches = sampler.getPatches();
51  		System.out.println("stop");
52  
53  		final double[][] data = new double[patches.size()][];
54  		for (int i = 0; i < data.length; i++)
55  			data[i] = patches.get(i).getDoublePixelVector();
56  
57  		// final PCAWhitening whitening = new PCAWhitening();
58  		final ZCAWhitening whitening = new ZCAWhitening(0.1, new PerExampleMeanCenter());
59  		whitening.train(data);
60  		final double[][] wd = whitening.whiten(data);
61  
62  		final SphericalKMeans skm = new SphericalKMeans(1600, 10);
63  		final SphericalKMeansResult res = skm.cluster(wd);
64  		final MBFImage tmp = new MBFImage(40 * (8 + 1) + 1, 40 * (8 + 1) + 1);
65  		tmp.fill(RGBColour.WHITE);
66  		for (int i = 0; i < 40; i++) {
67  			for (int j = 0; j < 40; j++) {
68  				final MBFImage patch = new MBFImage(res.centroids[i * 40 + j], 8, 8, 3, false);
69  				tmp.drawImage(patch, i * (8 + 1) + 1, j * (8 + 1) + 1);
70  			}
71  		}
72  		tmp.subtractInplace(-1.5f);
73  		tmp.divideInplace(3f);
74  		DisplayUtilities.display(tmp);
75  	}
76  }