Learning to Remove Multipath Distortions in Time-of-Flight Range Images for a Robotic Arm Setup
Range images captured by Time-of-Flight (ToF) cameras are corrupted with multipath distortions due to interaction between modulated light signals and scenes. The interaction is often complicated, which makes a model-based solution elusive. We propose a learning-based approach for removing the multipath distortions for a ToF camera in a robotic arm setup. Our approach is based on deep learning. We use the robotic arm to automatically collect a large amount of ToF range images containing various multipath distortions. The training images are automatically labeled by leveraging a high precision structured light sensor available only in the training time. In the test time, we apply the learned model to remove the multipath distortions. This allows our robotic arm setup to enjoy the speed and compact form of the ToF camera without compromising with its range measurement errors. We conduct extensive experimental validations and compare the proposed method to several baseline algorithms. The experiment results show that our method achieves 55% error reduction in range estimation and largely outperforms the baseline algorithms.
Time-of-flight (ToF) cameras capture scene depth by measuring phase delay of a modulated signal emitted from an infrared LED traveling in the space. Compared to a structured light sensor or a laser scanner, ToF cameras have the advantages of low cost, high speed, and compact form. These are important system parameters for the robotic arm setup considered in the paper, which consists of a robotic arm and a range sensor. A compact sensor in the robotic arm is less likely to collide with other objects during operation. The high speed feature allows objects to remain in motion during imaging.
Despite the advantages, ToF cameras are not yet widely deployed in real world applications. This is mainly because range measurements from ToF cameras are error-prone. In addition to random noise, ToF range measurements also suffer from intrinsic systematic errors and multipath distortions. The intrinsic systematic errors are due to the limitations in manufacturing the ToF camera hardware, including the infrared emitter (non-ideal sinusoidal modulation), the sensor (non-identical CMOS gates), and the optics (radial distortion and vignetting effects) [10, 23]. The systematic errors result in measurement bias but can be reduced by calibration [14, 21, 25, 26]. On the other hand, the multipath distortions are the result of interaction between light signals and scenes; they include 1) range over-shooting distortions due to the superposition of the reflection light signals from nearby structures, and 2) range over-smoothing distortions due to the superposition of the reflection light signals from the foreground and background objects as illustrated in Figure 1.
We propose a learning-based approach for removing the multipath distortions in range images captured by a ToF camera in a robotic arm setup. During training, we learn the multipath distortion patterns. The training data are automatically collected and labeled using the robotic arm and a high precision structured light sensor available only during training. After training, we apply the learned model to remove the multipath distortions in the ToF range images. This allows our robotic arm setup to enjoy the speed and compact form of the ToF camera without compromising with the range measurement errors. Our approach is based on two neural networks (NNs) where one NN learns a mapping from the ToF measurements to the true range values, while the other learns to detect object boundaries for guiding the geometry content propagation in a geodesic filtering framework . We conduct extensive experimental validations and compare our method with several baseline approaches. The results show that our method achieves more than 55% noise reduction and largely outperforms the baseline methods.
1.1 Related Work
Several methods based on ray tracing are proposed for removing multipath distortions in ToF range images [20, 16, 13, 15]. The distortion reduction process starts with an initial estimate of the true range image, which is then gradually refined using physics-based simulation. Preliminary results along this research line are shown on simple scenes consisting of several planar objects. Extending them to complex scenes is a non-trivial task. In addition, ray tracing are computationally expensive.
Setups utilizing multiple modulation frequencies are proposed for removing the multipath distortions in [22, 3, 12, 17, 9]. These methods require special hardware and, hence, are inapplicable to off-the-shelf ToF cameras where only one modulation frequency is available. Stereo ToF camera  is proposed to improve the range measurements, but it requires two ToF cameras and a baseline, which makes the setup non-compact. Our method does not require special hardware and is directly applicable to off-the-shelf ToF cameras.
Along with the range image, ToF cameras also capture an amplitude image. The amplitude image records the strength of the returning light per pixel, which quantifies the signal-to-noise ratio for each pixel. It can be used as a confidence measure. Frank et al.  propose a filtering method where the filter shape is determined by the amplitude-based confidence measure. However, the amplitude-based confidence measure itself can be corrupted with the multipath distortions. Reynolds et al.  propose learning a confidence measure using a random forest regressor based on hand-crafted features.
Lenzen et al.  propose a Total Variation (TV) method for ToF range image denoising. It estimates the true range image by minimizing a predefined objective function that consists of data-fidelity terms and regularization terms. Although the method is effective in reducing random noise, it does not model multipath distortions and tends to smooth out fine structures in the range images.
Recently, deep neural networks are applied for image processing tasks including image denoising  and superresolution . However, since ToF range images have very different characteristics to conventional intensity images, these approaches are not directly applicable to removing the multipath distortions in ToF range images.
The contributions of the paper are summarized below.
We propose a learning-based approach for removing the multipath distortions in range images captured by Time-of-Flight cameras where the training data are automatically collected and labeled using a robotic arm setup.
Our approach is based on geodesic filtering and two NNs where one NN learns a mapping from the ToF measurements to the true range values, while the other learns to detect object boundaries for guiding the geometry content propagation in the geodesic filtering.
We conduct extensive experimental validations and compare the proposed method to several baseline algorithms. The results show that the proposed method achieves 55% noise reduction and significantly outperforms the baseline methods.
We consider a setup where a ToF camera is mounted on a robotic arm as illustrated in Figure 2. In the training phase, we use a high precision structured light sensor to compute a reference (ground truth) range image for each ToF range image. We use the robotic setup to efficiently capture a large labeled training dataset for learning the model. After training, the structured light sensor is dismounted and we apply the learned model to remove the multipath distortions in the ToF range image. The setup has several benefits including:
It leverages the range measurement
1from the structured light sensor to compute the reference range values, avoiding expensive human annotation.
It can effortlessly capture a large amount of labeled data by automatically positioning the robotic arm at different viewpoints.
In the test phase, we directly apply the learned model for removing the multipath distortions in the ToF range image. It enjoys the compact form and high speed of the ToF camera without suffering from its error-prone range measurements.
3 Removing Multipath Distortions
Our method is based on feed-forward NNs. It consists of two NNs, referred to as and , where defines a mapping from the ToF range measurements to the true range values, while learns to detect the true object boundary locations for guiding the geometry content propagation. Ideally, alone should be sufficient to recover the true range measurements. However, we found that although has good performance in the smooth region, it has difficulty in the object boundary region. Hence, we use in conjunction with a geodesic filtering algorithm to propagate the recovered range measurements from the smooth region to the boundary region, which improves the range estimation accuracy in the boundary region. The proposed method is summarized in Figure 3.
The proposed method computes the final range value at pixel as
where is the estimated range value at pixel from applying . The set denotes the neighbors of , and is a weighting function measuring the similarity between pixels and . The weighting function is given by
where is the geodesic distance between the two pixels. The kernel bandwidth is set to 2 in our experiments. We note that both the neighbor set and the weighting function depend on .
We use an edge map to compute the geodesic distances between pixels on the image grid. On the image grid, the distance between two neighboring pixels is set to a small constant if they are not in two sides of an edge; otherwise, it is set to . The geodesic distance is given by the length of the shortest path between the two pixels. In this way, we ensure that the pixels used to smooth a target pixel carry more weights if they are from the same surface as the target pixel. Computing pairwise geodesic distance is computationally expensive. We use the fast approximation algorithm introduced in  to compute the geodesic distances from a pixel to its nearest neighbors simultaneously. The constant is set to 81 in our experiments.
We assume that the multipath distortions in the ToF range images are mostly local—the contribution to the multipath distortions from remote objects is much less than that from nearby structures. For example, the distortion in the left side of the wood box in Figure 1 is mainly from the signal around the left side corner and is non-related to the signal around the right side corner of the wood box. Such an assumption can be easily satisfied for the assembly line configuration, which is a major application target of the robotic arm setup.
Based on the assumption, we adopt a patch-based representation. We use the measurement data in the image patch centered at a pixel to construct the input data for inferring the target quantity at the pixel for both and . Most of off-the-shelf ToF cameras capture an amplitude image in addition to the range image. The amplitude image represents the strength of the returning signals at each pixel. Both raw range and amplitude images are inputs to the NNs.
3.1 Range-Recovery NN
We learn a regression function that maps the ToF measurement data in the patch centered at the target pixel location to its range value. Let be the target pixel. The input data for each pixel is given by
The term is a vector encoding the range values of the pixels in the patch centered at . It is obtained by subtracting the range values by the range value of , followed by a normalization step, mapping the values to . The second term, , is obtained via applying a similar operation to the amplitude values in the patch. The patch size used in our experiments is . Hence, the dimensions of and are both 120.
The last two terms in (3), and ), are binary vectors encoding range and amplitude values of , which are subtracted to compute and . The dimensions of the binary vectors are set to 20 in our experiments. They are computed by uniformly quantizing the range and amplitude values into different intervals, respectively. We set the corresponding element of the binary vector to if the range and amplitude values fall in the interval; otherwise 0.
The target value for the range-recovery NN is given by the difference between the ground truth range value and the ToF range value; that is
where and are the ground truth and the ToF range value at , respectively. We truncate the target value using a threshold, which is 15 mm in our experiments. The pair serves as the training data for learning .
The range-recovery NN is defined as a three-layer feed-forward network where the first layer contains neurons, and the second and third layer each contains neurons. The neurons are fully connected. We apply rectified linear units (ReLU) after the neurons for modeling the non-linear relationship between the input data and target value. We do not use the pooling layer popular in modern convolutional neural networks since pooling reduces the output resolution. The neurons in the third layer are fed into the network output neuron. We train through minimizing the Euclidean loss. We use the learned to recover the true range value. Note that the recovered range is given by
3.2 Boundary-Detection NN
We compute the ground truth boundaries and their directions by applying the Canny edge detector to the reference range images obtained with the structured light sensor. We divide the detected edges into groups based on a uniform quantization of the edge orientation. For each group, the edges from the other groups as well as non-edge pixels are used as the negative training samples. They are used to learn edge detectors, each for one direction.
We use separate feed-forward NNs for the edge detectors. Similar to the case of the range-recovery NN, we extract the measurements surrounding a pixel within a patch to form an input vector
The structures for the boundary-detection NNs are identical. Each consists of two layers where the first layer has 40 neurons and the second layer has 20 neurons. The neurons are all fully connected. Each layer is followed by the ReLU nonlinearity. The output layers have two units, which represent the edge and non-edge likelihood scores after applying a softmax operation. The networks are trained via minimizing the cross entropy loss.
For each pixel, we find a maximum response from the NNs for the boundary likelihood score and direction, which is similar to the procedure described in . We apply the non-maximum suppression and hysteresis thresholding to compute the binary edge map for geodesic filtering as used in the Canny edge detector.
We used the robotic arm setup discussed in Section 2 to obtain labeled range images. We collected two datasets, one for training and the other for performance evaluation, which are called MD-Train and MD-Test, respectively
The precision of the structured light sensor was 0.5 mm. We performed the hand-eye calibration before the dataset collection. Hence, the coordinate transformation between the structured light sensor and the ToF camera was known. We used a measurement fusion approach to compute the reference range image for each ToF range image. Due to the baseline between the projector and camera, the structured light sensor failed to obtain the range measurements in the occluded area. This problem was mitigated by scanning the scene with 10 different in-plane rotation angles at 3 different distances from the objects. We transformed all the 30 scans to the ToF camera coordinate system. For each pixel in the ToF camera, we considered the 3D points projected to the pixel, found the cluster of 3D points closest to the camera center, and used the median of the range values of the 3D points in the cluster as the reference range value. If an insufficient number of 3D points were projected to a pixel, the reference range for the pixel was denoted as missing. This happened because the field of views of the sensors were different and because some regions were occluded even using the multiple viewpoints. The missing pixels were marked in the green color in the figures reported in the experiment section. The missing pixels were neither used for training nor for performance evaluation.
The MD-Test was captured in the same way. It contained 20 scenes constructed by randomly arranging a subset of 30 different objects in the platform. Note that none of the objects used in the MD-Train was used in the MD-Test. For each scene, we also captured ToF images from 18 different views, providing 360 images. We used the structured light sensor to compute the reference range image for the purpose of performance evaluation.
Our setup can be applied to a wide variety of off-the-shelf ToF cameras, such as those from Mesa Imaging, PMD, and SoftKinetic. In this paper, we used a SoftKinetic DS325 ToF camera, having a resolution of 320240
The 540 training images in the MD-Train generated about 10 million image patches. Hence, our NNs learned the input-output relationship from the 10 million training samples. We used the public Caffe library  to train our NNs. Stochastic gradient descent with momentum was used for optimization. The momentum weight was set to 0.9, while the batch size was set to 500. We trained the NNs for 40 epochs.
5.1 Baseline Methods
We compared our algorithm with several competing algorithms for removing ToF range distortions, including
The amplitude image was used in the JBF and WMF as the guidance image. The RF-WF used a weighted Gaussian filter. The weights in the RF-WF were determined by a random forest regressor. Note that  proposed using the random forest regressor to learn a confidence measure but did not consider removing range distortions in ToF range images. We applied the learned confidence measure from the random forest regressor  with the weighted filtering method proposed  to remove the multipath distortions in the ToF range image. We used the MD-Train dataset to train the random forest regressor. For the RFD, the random forest regressor  was directly trained to remove the distortion directly using the same training data as the proposed method. Comparison with the RFD justified the use of NNs for removing multipath distortion. For the TV , we used L1 norm for the regularization term in the objective function to preserve structure of the range image. The TV objective function was optimized by a primal-dual approach . The PD algorithm  was proposed for the ToF range image superresolution task. It formulated the superresolution as an inference problem in the Markov Random Field where the latent variables were the indices to the reference patches. The data term was the L2 distance between observations and the estimated patches, and the pairwise term was the difference between neighboring patches. We adapted the PD algorithm for the ToF range image distortion removal task. The performance of PD depended on the comprehensiveness of the training patches; however, the computational time also increased with the number of training patches. We used the pre-generated patches provided by the authors, which were designed to be generic for various scenes.
We used the implementations provided by the authors of the original papers whenever available. The competing algorithms all had free parameters. We performed grid search and used the best one to report the performance.
5.2 Performance Metric
We used the MD-Test for performance evaluation. The performance was measured by the percentage of correctly recovered range values. A range value was considered correctly recovered if its deviation from the reference range value was less than a threshold. We varied the threshold and plotted the performance curve. We computed the results for the whole image regions and for the boundary regions separately (5-pixel margin from the ground truth boundary pixels). A similar performance metric was used in the Middlebury stereo benchmark .
Figure 5 compares the performance of the competing algorithms. The raw ToF range images were noisy. For the threshold of 5mm, only about 30% of range measurements were correct. After applying the calibration for removing the intrinsic systematic error, the accuracy improved to 45%. The TV, PD, RF-WF, WMF, and JBF algorithms improved the measurements based on the results after the intrinsic calibration. However, the improvements were marginal. The RFD was ranked 2nd, which showed the benefit of a learning-based algorithm for distortion removal. The proposed algorithm modeled the multipath distortions using the high capacity NNs and achieved the best performance. It rendered an accuracy of 63% at 5mm threshold, corresponding to a 55% improvement with respect to the raw measurements.
Figure 6 shows qualitative comparisons. We observed that the competing algorithms were able to filter out the random noise but failed to correct the multipath distortions in the boundary regions. The results still contained severe range over-shooting errors in the boundary regions as indicated by the red color regions in the figure. In contrast, the proposed algorithm removed a large portion of the multi-path distortions. Detailed comparisons in the boundary regions are provided in Figures 7 and 8. We found that our algorithm largely removed the error in the boundary regions, while the competing algorithms tended to blur the boundaries.
5.4 Algorithm Component Analysis
Our method has two components: 1) the range-recovery NN and 2) the geodesic filtering using the boundary map computed by the boundary-detection NN. We analyzed the contributions from the two components individually using the MD-Test. Figure 9 shows the performance curves of the results in the whole image and boundary regions. We found that a large portion of performance improvements were due to the range-recovery NN. Using the geodesic filtering alone led to slight improvement. However, the improvement were mostly in the boundary regions, which complimented the results from the range-recovery NN. Combining the two components, our method achieved large performance boost in both smooth and boundary regions.
5.5 Performance of Boundary-Detection NN
Figure 10 compares boundary detection results by our method (red) and by the Canny detector (black). The edges detected by the Canny detector were off (5 pixels in average error) from the reference edge locations. Edges obtained with our method were much closer to the reference (blue) edges. They often exactly overlapped with each other (magenta), whereas the Canny edges were usually far away from the ground truth. Figure 11 shows precision and recall curves of the edge detectors evaluated with the MD-Test, which demonstrates that our edges were significantly more accurate than the Canny edges. The accurate edge localization can be useful for edge-based object pose estimation algorithms such as [32, 7].
The time required for training each NN was about 20 minutes. The computation of the range-recovery NN was about 35 ms using a GPU card, while that of the boundary-detection NN took about 30 ms. The geodesic filtering algorithm was performed on CPU, which took about 100 ms. The overall computation time was 165 ms. The experiments were conducted on a 4-core Intel i7 processor with an Nvidia Titan graphic card.
We introduced a learning-based approach for removing the multipath distortions in ToF range images, which is based on deep learning and a geodesic filtering algorithm. Experiment results showed that our approach effectively reduced the range over-shooting and range over-smoothing distortions. ToF range measurements are also known to correlate with object material property. In future, we plan to study the feasibility of using the learning-based approach for addressing the material property dependency.
The authors would like to thank Jay Thornton, John Barnwell, and Yukiyasu Domae for their support and feedback.
- In practice, we use a fusion approach that combines several range images captured by the structure light sensor for obtaining more accurate range measurments.
- The MD (Multipath Distortion) datasets are available in https://sites.google.com/site/kilhoson/learning_tof
- The SoftKinetic DS325 ToF camera also has a color camera. However, it was used only for visualization purpose in Figures 4 and 6 and not for computation.
- O. M. Aodha, N. D. Campbell, A. Nair, and G. J. Brostow. Patch based synthesis for single depth image super-resolution. In ECCV, 2012.
- P. Arbelaez, M. Maire, C. Fowlkes, and J. Malik. Contour detection and hierarchical image segmentation. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011.
- A. Bhandari, A. Kadambi, R. Whyte, C. Barsi, M. Feigin, A. Dorrington, and R. Raskar. Resolving multipath interference in time-of-flight imaging via modulation frequency diversity and sparse regularization. Optics Letters, 2014.
- H. C. Burger, C. J. Schuler, and S. Harmeling. Image denoising: Can plain neural networks compete with BM3D? In CVPR, 2012.
- V. Castaneda, D. Mateus, and N. Navab. Stereo time-of-flight. In ICCV, 2011.
- A. Chambolle and T. Pock. A first-order primal-dual algorithm for convex problems with applications to imaging. Journal of Mathematical Imaging and Vision, 2011.
- C. Choi, Y. Taguchi, O. Tuzel, M.-Y. Liu, and S. Ramalingam. Voting-based pose estimation for robotic assembly using a 3D sensor. In ICRA, 2012.
- C. Dong, C. Loy, K. He, and X. Tang. Learning a deep convolutional network for image super-resolution. In ECCV, 2014.
- A. Dorrington, J. Godbaz, M. Cree, A. Payne, and L. Streeter. Separating true range measurements from multi-path and scattering interference in commercial range cameras. In IS&T/SPIE Electronic Imaging, 2011.
- S. Foix, G. Alenya, and C. Torras. Lock-in time-of-flight (ToF) cameras: A survey. IEEE Sensors Journal, 2011.
- M. Frank, M. Plaue, and F. A. Hamprecht. Denoising of continuous-wave time-of-flight depth images using confidence measures. Optical Engineering, 2009.
- D. Freedman, Y. Smolin, E. Krupka, I. Leichter, and M. Schmidt. SRA: Fast removal of general multipath for ToF sensors. In ECCV, 2014.
- S. Fuchs. Multipath interference compensation in time-of-flight camera images. In ICPR, 2010.
- S. Fuchs and G. Hirzinger. Extrinsic and depth calibration of tof-cameras. In CVPR, 2008.
- S. Fuchs, M. Suppa, and O. Hellwich. Compensation for multipath in tof camera measurements supported by photometric calibration and environment integration. In Computer Vision Systems, 2013.
- J. P. Godbaz, M. J. Cree, and A. A. Dorrington. Closed-form inverses for the mixed pixel/multipath interference problem in AMCW lidar. In IS&T/SPIE Electronic Imaging, volume 8296, 2012.
- M. Gupta, S. K. Nayar, M. Hullin, and J. Martin. Phasor imaging: A generalization of correlation-based time-of-flight imaging. ACM Transactions on Graphics (TOG), 2015.
- K. He, J. Sun, and X. Tang. Guided image filtering. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2013.
- Y. Jia, E. Shelhamer, J. Donahue, S. Karayev, J. Long, R. Girshick, S. Guadarrama, and T. Darrell. Caffe: Convolutional architecture for fast feature embedding. arXiv preprint arXiv:1408.5093, 2014.
- D. Jimenez, D. Pizarro, M. Mazo, and S. Palazuelos. Modelling and correction of multipath interference in time of flight cameras. In CVPR, 2012.
- Y. M. Kim, D. Chan, C. Theobalt, and S. Thrun. Design and calibration of a multi-view tof sensor fusion system. In CVPR Workshop, 2008.
- A. Kirmani, A. Benedetti, and P. A. Chou. SPUMIC: Simultaneous phase unwrapping and multipath interference cancellation in time-of-flight cameras using spectral methods. In ICME, 2013.
- D. Lefloch, R. Nair, F. Lenzen, H. Schaefer, L. Streeter, M. J. Cree, R. Koch, and A. Kolb. Technical foundation and calibration methods for time-of-flight cameras. In Time-of-Flight and Depth Imaging. Sensors, Algorithms, and Applications, pages 3–24. 2013.
- F. Lenzen, K. Kim, H. SchÃÂ¤fer, R. Nair, S. Meister, F. Becker, C. Garbe, and C. Theobalt. Denoising strategies for time-of-flight data. In Time-of-Flight and Depth Imaging. Sensors, Algorithms, and Applications, pages 25–45. 2013.
- M. Lindner and A. Kolb. Lateral and depth calibration of pmd-distance sensors. In Advances in Visual Computing, pages 524–533. 2006.
- M. Lindner, I. Schiller, A. Kolb, and R. Koch. Time-of-flight sensor calibration for accurate range sensing. Computer Vision and Image Understanding, pages 1318–1328, 2010.
- M.-Y. Liu, O. Tuzel, and Y. Taguchi. Joint geodesic upsampling of depth images. In CVPR, 2013.
- Z. Ma, K. He, Y. Wei, J. Sun, and E. Wu. Constant time weighted median filtering for stereo matching and beyond. In ICCV, 2013.
- G. Petschnigg, R. Szeliski, M. Agrawala, M. Cohen, H. Hoppe, and K. Toyama. Digital photography with flash and no-flash image pairs. ACM Transactions on Graphics (TOG), 2004.
- M. Reynolds, J. Dobos, L. Peel, T. Weyrich, and G. Brostow. Capturing time-of-flight data with confidence. In CVPR, 2011.
- D. Scharstein and R. Szeliski. A taxonomy and evaluation of dense two-frame stereo correspondence algorithms. International Journal of Computer Vision, 2002.
- O. Tuzel, M.-Y. Liu, Y. Taguchi, and A. Raghunathan. Learning to rank 3D features. In ECCV, 2014.
- Q. Yang, N. Ahuja, R. Yang, K.-H. Tan, J. Davis, B. Culbertson, J. Apostolopoulos, and G. Wang. Fusion of median and bilateral filtering for range image upsampling. IEEE Transactions on Image Processing, 2013.
- Q. Yang, R. Yang, J. Davis, and D. Nister. Spatial-depth super resolution for range images. In CVPR, 2007.
- Q. Zhang, L. Xu, and J. Jia. 100+ times faster weighted median filter (wmf). In CVPR, 2014.