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.image.feature.global;
31  
32  import org.apache.commons.math.stat.descriptive.SummaryStatistics;
33  import org.openimaj.citation.annotation.Reference;
34  import org.openimaj.citation.annotation.ReferenceType;
35  import org.openimaj.feature.DoubleFV;
36  import org.openimaj.feature.FeatureVectorProvider;
37  import org.openimaj.image.FImage;
38  import org.openimaj.image.MBFImage;
39  import org.openimaj.image.analyser.ImageAnalyser;
40  import org.openimaj.image.mask.AbstractMaskedObject;
41  
42  /**
43   * Estimate the variation in saturation of an image using the RGB approximation
44   * of avg(max(R,G,B) - min(R,G,B)).
45   * 
46   * @author Jonathon Hare (jsh2@ecs.soton.ac.uk)
47   */
48  @Reference(
49  		type = ReferenceType.Inproceedings,
50  		author = { "Jose San Pedro", "Stefan Siersdorfer" },
51  		title = "Ranking and Classifying Attractiveness of Photos in Folksonomies",
52  		year = "2009",
53  		booktitle = "18th International World Wide Web Conference",
54  		pages = { "771", "", "771" },
55  		url = "http://www2009.eprints.org/78/",
56  		month = "April")
57  public class SaturationVariation extends AbstractMaskedObject<FImage>
58  		implements
59  		ImageAnalyser<MBFImage>,
60  		FeatureVectorProvider<DoubleFV>
61  {
62  	double saturationVariation;
63  
64  	@Override
65  	public DoubleFV getFeatureVector() {
66  		return new DoubleFV(new double[] { saturationVariation });
67  	}
68  
69  	@Override
70  	public void analyseImage(MBFImage image) {
71  		final FImage R = image.getBand(0);
72  		final FImage G = image.getBand(1);
73  		final FImage B = image.getBand(2);
74  
75  		final SummaryStatistics stats = new SummaryStatistics();
76  
77  		if (mask != null) {
78  			for (int y = 0; y < R.height; y++) {
79  				for (int x = 0; x < R.width; x++) {
80  					if (mask.pixels[y][x] == 1) {
81  						final float min = Math.min(Math.min(R.pixels[y][x], G.pixels[y][x]), B.pixels[y][x]);
82  						final float max = Math.max(Math.max(R.pixels[y][x], G.pixels[y][x]), B.pixels[y][x]);
83  						stats.addValue(max - min);
84  					}
85  				}
86  			}
87  		} else {
88  			for (int y = 0; y < R.height; y++) {
89  				for (int x = 0; x < R.width; x++) {
90  					final float min = Math.min(Math.min(R.pixels[y][x], G.pixels[y][x]), B.pixels[y][x]);
91  					final float max = Math.max(Math.max(R.pixels[y][x], G.pixels[y][x]), B.pixels[y][x]);
92  					stats.addValue(max - min);
93  				}
94  			}
95  		}
96  
97  		saturationVariation = stats.getStandardDeviation();
98  	}
99  
100 	/**
101 	 * Get the variation in saturation of the last image processed with
102 	 * {@link #analyseImage(MBFImage)}.
103 	 * 
104 	 * @return the saturation
105 	 */
106 	public double getSaturationVariation() {
107 		return saturationVariation;
108 	}
109 }