All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.openimaj.image.processing.algorithm.AnisotropicDiffusion Maven / Gradle / Ivy

/**
 * Copyright (c) 2011, The University of Southampton and the individual contributors.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 *   * 	Redistributions of source code must retain the above copyright notice,
 * 	this list of conditions and the following disclaimer.
 *
 *   *	Redistributions in binary form must reproduce the above copyright notice,
 * 	this list of conditions and the following disclaimer in the documentation
 * 	and/or other materials provided with the distribution.
 *
 *   *	Neither the name of the University of Southampton nor the names of its
 * 	contributors may be used to endorse or promote products derived from this
 * 	software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package org.openimaj.image.processing.algorithm;

import org.openimaj.citation.annotation.Reference;
import org.openimaj.citation.annotation.ReferenceType;
import org.openimaj.image.FImage;
import org.openimaj.image.pixel.ConnectedComponent;
import org.openimaj.image.pixel.ConnectedComponent.ConnectMode;
import org.openimaj.image.processor.SinglebandImageProcessor;

import net.jafama.FastMath;

/**
 * Implementation of Perona & Malik's image filtering by anisotropic
 * diffusion. Enables edge-preserving image smoothing.
 *
 * @author Jonathon Hare ([email protected])
 */
@Reference(
		type = ReferenceType.Article,
		author = { "Perona, P.", "Malik, J." },
		title = "Scale-Space and Edge Detection Using Anisotropic Diffusion",
		year = "1990",
		journal = "IEEE Trans. Pattern Anal. Mach. Intell.",
		pages = { "629", "", "639" },
		url = "http://dx.doi.org/10.1109/34.56205",
		month = "July",
		number = "7",
		publisher = "IEEE Computer Society",
		volume = "12",
		customData = {
				"issue_date", "July 1990",
				"issn", "0162-8828",
				"numpages", "11",
				"doi", "10.1109/34.56205",
				"acmid", "78304",
				"address", "Washington, DC, USA"
		})
public class AnisotropicDiffusion implements SinglebandImageProcessor {
	/**
	 * Interface describing a function that computes the conduction coefficient as a
	 * function of space and gradient magnitude.
	 *
	 * @author Jonathon Hare ([email protected])
	 *
	 */
	public static interface ConductionCoefficientFunction {
		/**
		 * Compute the conduction coefficient given gradient and position.
		 *
		 * @param grad
		 *            the gradient
		 * @param x
		 *            the x-position in the image
		 * @param y
		 *            the y-position in the image
		 * @return the conduction coefficient
		 */
		float apply(float grad, int x, int y);
	}

	/**
	 * The first conduction function proposed by Perona & Malik, that privileges
	 * high contrast edges over low constrast ones.
	 *
	 * 
	 * g(∇I) = exp( - (||∇I/κ||²) )
	 * 
* * @author Jonathon Hare ([email protected]) */ public static class HighConstrastEdgeFunction implements ConductionCoefficientFunction { private float kappa; /** * Construct with the given kappa value * * @param kappa * kappa */ public HighConstrastEdgeFunction(float kappa) { this.kappa = kappa; } @Override public float apply(float val, int x, int y) { final float t = val / kappa; return (float) FastMath.exp(-t * t); } }; /** * The second conduction function proposed by Perona & Malik, that * privileges wider regions over smaller ones. * *
	 * g(∇I) = 1 / ( 1 + (||∇I/κ||²) )
	 * 
* * @author Jonathon Hare ([email protected]) */ public static class WideRegionFunction implements ConductionCoefficientFunction { private float kappa; /** * Construct with the given kappa value * * @param kappa * kappa */ public WideRegionFunction(float kappa) { this.kappa = kappa; } @Override public float apply(float val, int x, int y) { final float t = val / kappa; return 1 / (1 + t * t); } }; protected int numIterations; protected float lambda = 1f / 7f; protected ConductionCoefficientFunction function; protected ConnectedComponent.ConnectMode neighbourMode; /** * Construct using the given parameters. * * @param numIterations * number of iterations of filtering to apply * @param lambda * the integration constant (less than or equal to 1/4 for * 4-neighbour mode and 1/8 for 8-neighbour) * @param function * the conduction coefficient function * @param neighbourMode * the neighbourhood mode */ public AnisotropicDiffusion(int numIterations, float lambda, ConductionCoefficientFunction function, ConnectMode neighbourMode) { this.numIterations = numIterations; this.lambda = lambda; this.function = function; this.neighbourMode = neighbourMode; } /** * Construct using the given parameters. 4-neighbour mode is used as per the * original paper. * * @param numIterations * number of iterations of filtering to apply * @param lambda * the integration constant (bigger than 0 and less than or equal to * 1/4) * @param function * the conduction coefficient function */ public AnisotropicDiffusion(int numIterations, float lambda, ConductionCoefficientFunction function) { this.numIterations = numIterations; this.lambda = lambda; this.function = function; this.neighbourMode = ConnectMode.CONNECT_4; } @Override public void processImage(FImage image) { for (int i = 0; i < numIterations; i++) { processImageOneIteration(image); } } /** * Perform a single iteration of anisotropic diffusion * * @param image * the image */ public void processImageOneIteration(FImage image) { switch (neighbourMode) { case CONNECT_4: processImageOneIteration4(image); case CONNECT_8: processImageOneIteration8(image); } } private void processImageOneIteration4(FImage image) { final float[][] tmp = image.clone().pixels; for (int y = 0; y < image.height; y++) { final int ym = Math.max(y - 1, 0); final int yp = Math.min(y + 1, image.height - 1); for (int x = 0; x < image.width; x++) { final int xm = Math.max(x - 1, 0); final int xp = Math.min(x + 1, image.width - 1); final float dN = tmp[ym][x] - tmp[y][x]; final float dS = tmp[yp][x] - tmp[y][x]; final float dE = tmp[y][xm] - tmp[y][x]; final float dW = tmp[y][xp] - tmp[y][x]; final float cN = function.apply(Math.abs(dN), x, y); final float cS = function.apply(Math.abs(dS), x, y); final float cE = function.apply(Math.abs(dE), x, y); final float cW = function.apply(Math.abs(dW), x, y); image.pixels[y][x] += lambda * (cN * dN + cS * dS + cE * dE + cW * dW); } } } private void processImageOneIteration8(FImage image) { final float[][] tmp = image.clone().pixels; final float wt = 0.5f; for (int y = 0; y < image.height; y++) { final int ym = Math.max(y - 1, 0); final int yp = Math.min(y + 1, image.height - 1); for (int x = 0; x < image.width; x++) { final int xm = Math.max(x - 1, 0); final int xp = Math.min(x + 1, image.width - 1); final float dN = tmp[ym][x] - tmp[y][x]; final float dS = tmp[yp][x] - tmp[y][x]; final float dE = tmp[y][xm] - tmp[y][x]; final float dW = tmp[y][xp] - tmp[y][x]; final float dNE = tmp[ym][xm] - tmp[y][x]; final float dSE = tmp[yp][xm] - tmp[y][x]; final float dSW = tmp[ym][xp] - tmp[y][x]; final float dNW = tmp[ym][xp] - tmp[y][x]; final float cN = function.apply(Math.abs(dN), x, y); final float cS = function.apply(Math.abs(dS), x, y); final float cE = function.apply(Math.abs(dE), x, y); final float cW = function.apply(Math.abs(dW), x, y); final float cNE = function.apply(Math.abs(dNE), x, y); final float cSE = function.apply(Math.abs(dSE), x, y); final float cSW = function.apply(Math.abs(dSW), x, y); final float cNW = function.apply(Math.abs(dNW), x, y); image.pixels[y][x] += lambda * (cN * dN + cS * dS + cE * dE + cW * dW + wt * (cNE * dNE + cSE * dSE + cSW * dSW + cNW * dNW)); } } } // public static void main(String[] args) throws Exception { // final MBFImage img = ImageUtilities.readMBF(new // File("/Users/jon/veg.jpg")); // final AnisotropicDiffusion ad = new AnisotropicDiffusion(20, 0.1f, new // HighConstrastEdgeFunction(30f / 255)); // // DisplayUtilities.display(img); // DisplayUtilities.display(img.process(ad)); // } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy