Example Point Cloud Depth Image

From BoofCV
Revision as of 23:00, 27 March 2016 by Peter (talk | contribs)
Jump to navigationJump to search

This example demonstrates how to create a 3D point cloud from a RGB-D sensor, such as the Kinect, and visualize it. RGB-D sensors have both visual and depth information. In this example the depth information is stored in a 16-bit image and the visual image in a standard color image. Calibration matching RGB an depth pixels to each other has already been done by the sensor.

Example Code:

Concepts:

  • RGB-D
  • Point clouds

Relevant Applets:

Related Examples:

Example Code

/**
 * Example of how to create a point cloud from a RGB-D (Kinect) sensor.  Data is loaded from two files, one for the
 * visual image and one for the depth image.
 *
 * @author Peter Abeles
 */
public class ExampleDepthPointCloud {

	public static void main( String args[] ) throws IOException {
		String nameRgb = UtilIO.pathExample("kinect/basket/basket_rgb.png");
		String nameDepth = UtilIO.pathExample("kinect/basket/basket_depth.png");
		String nameCalib = UtilIO.pathExample("kinect/basket/visualdepth.xml");

		VisualDepthParameters param = UtilIO.loadXML(nameCalib);

		BufferedImage buffered = UtilImageIO.loadImage(nameRgb);
		Planar<GrayU8> rgb = ConvertBufferedImage.convertFromMulti(buffered,null,true,GrayU8.class);
		GrayU16 depth =
				ConvertBufferedImage.convertFrom(UtilImageIO.loadImage(nameDepth),null,GrayU16.class);

		FastQueue<Point3D_F64> cloud = new FastQueue<Point3D_F64>(Point3D_F64.class,true);
		FastQueueArray_I32 cloudColor = new FastQueueArray_I32(3);

		VisualDepthOps.depthTo3D(param.visualParam, rgb, depth, cloud, cloudColor);

		DenseMatrix64F K = PerspectiveOps.calibrationMatrix(param.visualParam,null);

		PointCloudViewer viewer = new PointCloudViewer(K, 15);
		viewer.setPreferredSize(new Dimension(rgb.width,rgb.height));

		for( int i = 0; i < cloud.size; i++ ) {
			Point3D_F64 p = cloud.get(i);
			int[] color = cloudColor.get(i);
			int c = (color[0] << 16 ) | (color[1] << 8) | color[2];
			viewer.addPoint(p.x,p.y,p.z,c);
		}

		// ---------- Display depth image
		// use the actual max value in the image to maximize its appearance
		int maxValue = ImageStatistics.max(depth);
		BufferedImage depthOut = VisualizeImageData.disparity(depth, null, 0, maxValue, 0);
		ShowImages.showWindow(depthOut,"Depth Image");

		// ---------- Display colorized point cloud
		ShowImages.showWindow(viewer,"Point Cloud");
		System.out.println("Total points = "+cloud.size);
	}
}