Brontekst bekijken van Research and Development 1/^Archief/2008-2009/Gezichts detectie/Onderzoek/NNBFD/bestanden/Preprocessor.java

<source lang=java>
/**
 * 
 */
package nl.ru.rd.facedetection.nnbfd;

import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.awt.image.Raster;

/**
 * A preprocessor used in the NNBFD for images.
 * 
 * Converts the image to grayscale, stores it in a matrix, and uses histogram and linear equalization.
 * 
 * @author Christiaan Thijssen s0814970
 * @author Kevin Reintjes s0814954
 * @author Wouter Geraedts (s0814857 - wgeraedts) PGP 66AA5935
 */
public class Preprocessor
{
	private HistogramEqualizer histogramEq = new HistogramEqualizer();

	/**
	 * Converts the image to grayscale, stores it in a matrix, and uses histogram and linear equalization.
	 * 
	 * @param colorImage
	 *            The image to be converted.
	 * @return A matrix with the applied conversion.
	 */
	public Matrix process(BufferedImage colorImage)
	{
		Matrix matrix = this.toMatrix(colorImage);

		this.histogramEq.equalize(matrix);

		return matrix;
	}

	private Matrix toMatrix(BufferedImage colorImage)
	{
		int height = colorImage.getHeight();
		int width = colorImage.getWidth();

		BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_BYTE_GRAY);
		Graphics g = image.getGraphics();
		g.drawImage(colorImage, 0, 0, null);
		g.dispose();

		Raster raster = image.getData();

		Matrix matrix = new Matrix(width, height);
		for(int y = 0; y < height; y++)
			for(int x = 0; x < width; x++)
				matrix.setValue(x, y, (short) raster.getSample(x, y, 0));

		return matrix;
	}
}

</source>