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