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.segmentation; 31 32 import java.util.List; 33 34 import org.openimaj.image.MBFImage; 35 import org.openimaj.image.colour.ColourSpace; 36 import org.openimaj.image.colour.RGBColour; 37 import org.openimaj.image.pixel.ConnectedComponent; 38 import org.openimaj.image.pixel.PixelSet; 39 import org.openimaj.image.processor.connectedcomponent.render.BlobRenderer; 40 41 /** 42 * Some utility functions for dealing with segmented output 43 * 44 * @author Jonathon Hare (jsh2@ecs.soton.ac.uk) 45 * 46 */ 47 public class SegmentationUtilities { 48 private SegmentationUtilities() { 49 } 50 51 /** 52 * Render the components to the image with randomly assigned colours. 53 * 54 * @param image 55 * Image to draw to 56 * @param components 57 * the components 58 * @return the image 59 */ 60 public static MBFImage renderSegments(MBFImage image, List<? extends PixelSet> components) { 61 for (final PixelSet cc : components) { 62 final BlobRenderer<Float[]> br = new BlobRenderer<Float[]>(image, RGBColour.randomColour()); 63 br.process(new ConnectedComponent(cc.pixels)); 64 } 65 66 return image; 67 } 68 69 /** 70 * Render the components to an image with randomly assigned colours. 71 * 72 * @param width 73 * Width of image. 74 * @param height 75 * Height of image. 76 * @param components 77 * the components. 78 * @return the rendered image. 79 */ 80 public static MBFImage renderSegments(int width, int height, List<? extends PixelSet> components) { 81 return renderSegments(new MBFImage(width, height, ColourSpace.RGB), components); 82 } 83 }