arxiv: v2 [cs.cv] 29 Dec 2017

Similar documents
Light-Field Database Creation and Depth Estimation

Learning Pixel-Distribution Prior with Wider Convolution for Image Denoising

DEPTH FUSED FROM INTENSITY RANGE AND BLUR ESTIMATION FOR LIGHT-FIELD CAMERAS. Yatong Xu, Xin Jin and Qionghai Dai

LIGHT FIELD (LF) imaging [2] has recently come into

A Fuller Understanding of Fully Convolutional Networks. Evan Shelhamer* Jonathan Long* Trevor Darrell UC Berkeley in CVPR'15, PAMI'16

On the Recovery of Depth from a Single Defocused Image

Coded Aperture for Projector and Camera for Robust 3D measurement

Robust Light Field Depth Estimation for Noisy Scene with Occlusion


Convolutional Networks for Image Segmentation: U-Net 1, DeconvNet 2, and SegNet 3

CROSS-LAYER FEATURES IN CONVOLUTIONAL NEURAL NETWORKS FOR GENERIC CLASSIFICATION TASKS. Kuan-Chuan Peng and Tsuhan Chen

fast blur removal for wearable QR code scanners

Tiny ImageNet Challenge Investigating the Scaling of Inception Layers for Reduced Scale Classification Problems

Dappled Photography: Mask Enhanced Cameras for Heterodyned Light Fields and Coded Aperture Refocusing

Defocus Map Estimation from a Single Image

Computational Cameras. Rahul Raguram COMP

Modeling the calibration pipeline of the Lytro camera for high quality light-field image reconstruction

Coded Aperture Pairs for Depth from Defocus

Simulated Programmable Apertures with Lytro

Understanding Neural Networks : Part II

Research on Hand Gesture Recognition Using Convolutional Neural Network

Recent Advances in Image Deblurring. Seungyong Lee (Collaboration w/ Sunghyun Cho)

Implementation of Adaptive Coded Aperture Imaging using a Digital Micro-Mirror Device for Defocus Deblurring

Lecture 23 Deep Learning: Segmentation

DYNAMIC CONVOLUTIONAL NEURAL NETWORK FOR IMAGE SUPER- RESOLUTION

arxiv: v2 [cs.cv] 29 Aug 2017

Coded Computational Photography!

Fast Non-blind Deconvolution via Regularized Residual Networks with Long/Short Skip-Connections

Synthetic View Generation for Absolute Pose Regression and Image Synthesis: Supplementary material

Restoration of Motion Blurred Document Images

A moment-preserving approach for depth from defocus

multiframe visual-inertial blur estimation and removal for unmodified smartphones

A Review over Different Blur Detection Techniques in Image Processing

Computational Approaches to Cameras

LENSLESS IMAGING BY COMPRESSIVE SENSING

Toward Non-stationary Blind Image Deblurring: Models and Techniques

Learning to Predict Indoor Illumination from a Single Image. Chih-Hui Ho

Coded photography , , Computational Photography Fall 2018, Lecture 14

Dynamic Scene Deblurring Using Spatially Variant Recurrent Neural Networks

Deconvolution , , Computational Photography Fall 2018, Lecture 12

To Denoise or Deblur: Parameter Optimization for Imaging Systems

Project Title: Sparse Image Reconstruction with Trainable Image priors

Lecture 18: Light field cameras. (plenoptic cameras) Visual Computing Systems CMU , Fall 2013

Colorful Image Colorizations Supplementary Material

arxiv: v3 [cs.cv] 18 Dec 2018

Single Digital Image Multi-focusing Using Point to Point Blur Model Based Depth Estimation

Lecture 19: Depth Cameras. Kayvon Fatahalian CMU : Graphics and Imaging Architectures (Fall 2011)

Deblurring. Basics, Problem definition and variants

Biologically Inspired Computation

Depth from Combining Defocus and Correspondence Using Light-Field Cameras

Coded photography , , Computational Photography Fall 2017, Lecture 18

Removing Temporal Stationary Blur in Route Panoramas

Introduction to Machine Learning

Time-Lapse Light Field Photography With a 7 DoF Arm

Deconvolution , , Computational Photography Fall 2017, Lecture 17

Non-Uniform Motion Blur For Face Recognition

Changyin Zhou. Ph.D, Computer Science, Columbia University Oct 2012

NU-Net: Deep Residual Wide Field of View Convolutional Neural Network for Semantic Segmentation

Optimal Camera Parameters for Depth from Defocus

Image Deblurring with Blurred/Noisy Image Pairs

Convolutional Neural Network-Based Infrared Image Super Resolution Under Low Light Environment

Light field sensing. Marc Levoy. Computer Science Department Stanford University

Pattern Recognition 44 (2011) Contents lists available at ScienceDirect. Pattern Recognition. journal homepage:

Performance Evaluation of Different Depth From Defocus (DFD) Techniques

Semantic Segmentation in Red Relief Image Map by UX-Net

Coded Aperture and Coded Exposure Photography

Accelerating defocus blur magnification

Coded Aperture Flow. Anita Sellent and Paolo Favaro

TRANSFORMING PHOTOS TO COMICS USING CONVOLUTIONAL NEURAL NETWORKS. Tsinghua University, China Cardiff University, UK

SURVEILLANCE SYSTEMS WITH AUTOMATIC RESTORATION OF LINEAR MOTION AND OUT-OF-FOCUS BLURRED IMAGES. Received August 2008; accepted October 2008

23270: AUGMENTED REALITY FOR NAVIGATION AND INFORMATIONAL ADAS. Sergii Bykov Technical Lead Machine Learning 12 Oct 2017

Design of Temporally Dithered Codes for Increased Depth of Field in Structured Light Systems

Depth from Diffusion

A Novel Image Deblurring Method to Improve Iris Recognition Accuracy

Depth Estimation Algorithm for Color Coded Aperture Camera

Detection and Segmentation. Fei-Fei Li & Justin Johnson & Serena Yeung. Lecture 11 -

Spline wavelet based blind image recovery

Image and Depth from a Single Defocused Image Using Coded Aperture Photography

arxiv: v1 [cs.lg] 2 Jan 2018

Capturing Light. The Light Field. Grayscale Snapshot 12/1/16. P(q, f)

Total Variation Blind Deconvolution: The Devil is in the Details*

Image Denoising using Dark Frames

Introduction to Video Forgery Detection: Part I

A Mathematical model for the determination of distance of an object in a 2D image

CS6670: Computer Vision

Semantic Segmentation on Resource Constrained Devices

Demosaicing and Denoising on Simulated Light Field Images

Accurate Disparity Estimation for Plenoptic Images

Single Camera Catadioptric Stereo System

Learning to Understand Image Blur

Evolving Measurement Regions for Depth from Defocus

An Introduction to Convolutional Neural Networks. Alessandro Giusti Dalle Molle Institute for Artificial Intelligence Lugano, Switzerland

Fast and High-Quality Image Blending on Mobile Phones

Lesson 08. Convolutional Neural Network. Ing. Marek Hrúz, Ph.D. Katedra Kybernetiky Fakulta aplikovaných věd Západočeská univerzita v Plzni.

Selective Detail Enhanced Fusion with Photocropping

Understanding camera trade-offs through a Bayesian analysis of light field projections - A revision Anat Levin, William Freeman, and Fredo Durand

Can you tell a face from a HEVC bitstream?

IMAGE TAMPERING DETECTION BY EXPOSING BLUR TYPE INCONSISTENCY. Khosro Bahrami and Alex C. Kot

Burst Photography! EE367/CS448I: Computational Imaging and Display! stanford.edu/class/ee367! Lecture 7! Gordon Wetzstein! Stanford University!

Coding and Modulation in Cameras

Transcription:

A Learning-based Framework for Hybrid Depth-from-Defocus and Stereo Matching Zhang Chen 1, Xinqing Guo 2, Siyuan Li 1, Xuan Cao 1 and Jingyi Yu 1 arxiv:1708.00583v2 [cs.cv] 29 Dec 2017 1 ShanghaiTech University, Shanghai, China. {chenzhang, lisy1, caoxuan, yujingyi}@shanghaitech.edu.cn 2 University of Delaware, Newark, DE, USA. xinqing@udel.edu Abstract Depth from defocus (DfD) and stereo matching are two most studied passive depth sensing schemes. The techniques are essentially complementary: DfD can robustly handle repetitive textures that are problematic for stereo matching whereas stereo matching is insensitive to defocus blurs and can handle large depth range. In this paper, we present a unified learning-based technique to conduct hybrid DfD and stereo matching. Our input is image triplets: a stereo pair and a defocused image of one of the stereo views. We first apply depth-guided light field rendering to construct a comprehensive training dataset for such hybrid sensing setups. Next, we adopt the hourglass network architecture to separately conduct depth inference from DfD and stereo. Finally, we exploit different connection methods between the two separate networks for integrating them into a unified solution to produce high fidelity 3D disparity maps. Comprehensive experiments on real and synthetic data show that our new learning-based hybrid 3D sensing technique can significantly improve accuracy and robustness in 3D reconstruction. 1. Introduction Acquiring 3D geometry of the scene is a key task in computer vision. Applications are numerous, from classical object reconstruction and scene understanding to the more recent visual SLAM and autonomous driving. Existing approaches can be generally categorized into active or passive 3D sensing. Active sensing techniques such as LIDAR and structured light offer depth map in real time but require complex and expensive imaging hardware. Alternative passive scanning systems are typically more cost-effective and can conduct non-intrusive depth measurements but maintaining its robustness and reliability remains challenging. These authors contribute to the work equally. Stereo matching and depth from defocus (DfD) are the two best-known passive depth sensing techniques. Stereo recovers depth by utilizing parallaxes of feature points between views. At its core is correspondences matching between feature points and patching the gaps by imposing specific priors, e.g., induced by the Markov Random Field. DfD, in contrast, infers depth by analyzing blur variations at same pixel captured with different focus settings (focal depth, apertures, etc). Neither technique, however, is perfect on its own: stereo suffers from ambiguities caused by repetitive texture patterns and fails on edges lying along epipolar lines whereas DfD is inherently limited by the aperture size of the optical system. It is important to note that DfD and stereo are complementary to each other: stereo provides accurate depth estimation even for distant objects whereas DfD can reliably handle repetitive texture patterns. In computational imaging, a number of hybrid sensors have been designed to combine the benefits of the two. In this paper, we seek to leverage deep learning techniques to infer depths in such hybrid DfD and stereo setups. Recent advances in neural network have revolutionized both high-level and low-level vision by learning a non-linear mapping between the input and output. Yet most existing solutions have exploited only stereo cues [23, 41, 42] and very little work addresses using deep learning for hybrid stereo and DfD or even DfD alone, mainly due to the lack of a fully annotated DfD dataset. In our setup, we adopt a three images setting: an allfocus stereo pair and a defocused image of one of the stereo views, the left in our case. We have physically constructed such a hybrid sensor by using Lytro Illum camera. We first generate a comprehensive training dataset for such an imaging setup. Our dataset is based on FlyingThings3D from [24], which contains stereo color pairs and ground truth disparity maps. We then apply occlusion-aware light field rendering[40] to synthesize the defocused image. Next, we

adopt the hourglass network [25] architecture to extract depth from stereo and defocus respectively. Hourglass network features a multi-scale architecture that consolidates both local and global contextures to output per-pixel depth. We use stacked hourglass network to repeat the bottom-up, topdown depth inferences, allowing for refinement of the initial estimates. Finally, we exploit different connection methods between the two separate networks for integrating them into a unified solution to produce high fidelity 3D depth maps. Comprehensive experiments on real and synthetic data show that our new learning-based hybrid 3D sensing technique can significantly improve accuracy and robustness in 3D reconstruction. 1.1. Related Work Learning based Stereo Stereo matching is probably one of the most studied problems in computer vision. We refer the readers to the comprehensive survey [31, 3]. Here we only discuss the most relevant works. Our work is motivated by recent advances in deep neural network. One stream focuses on learning the patch matching function. The seminal work by Žbontar and LeCun [43] leveraged convolutional neural network (CNN) to predict the matching cost of image patches, then enforced smoothness constraints to refine depth estimation. [41] investigated multiple network architectures to learn a general similarity function for wide baseline stereo. Han et al. [10] described a unified approach that includes both feature representation and feature comparison functions. Luo et al. [23] used a product layer to facilitate the matching process, and formulate the depth estimation as a multi-class classification problem. Other network architectures [5, 22, 27] have also been proposed to serve a similar purpose. Another stream of studies exploits end-to-end learning approach. Mayer et al. [24] proposed a multi-scale network with contractive part and expanding part for real-time disparity prediction. They also generated three synthetic datasets for disparity, optical flow and scene flow estimation. Knöbelreiter et al. [17] presented a hybrid CNN+CRF model. They first utilized CNNs for computing unary and pairwise cost, then feed the costs into CRF for optimization. The hybrid model is trained in an end-to-end fashion. In this paper, we employ end-to-end learning approach for depth inference due to its efficiency and compactness. Depth from Defocus The amount of blur at each pixel carries information about object s distance, which could benefit numerous applications, such as saliency detection [21, 20]. To recover scene geometry, earlier DfD techniques [33, 29, 39] rely on images captured with different focus settings (moving the objects, the lense or the sensor, changing the aperture size, etc). More recently, Favaro and Soatto [9] formulated the DfD problem as a forward diffusion process where the amount of diffusion depends on the depth of the scene. [18, 44] recovered scene depth and all-focused image from images captured by a camera with binary coded aperture. Based on a per-pixel linear constraint from image derivatives, Alexander et al. [1] introduced a monocular computational sensor to simultaneously recover depth and motion of the scene. Varying the size of the aperture [28, 8, 35, 2] has also been extensively investigated. This approach will not change the distance between the lens and sensor, thus avoiding the magnification effects. Our DfD setting uses a defocused and all-focused image pair as input, which can be viewed as a special case of the varying aperture. To tackle the task of DfD, we utilize a multi-scale CNN architecture. Different from conventional hand-crafted features and engineered cost functions, our data-driven approach is capable of learning more discriminative features from the defocus image and inferring the depth at a fraction of the computational cost. Hybrid Stereo and DfD Sensing In the computational imaging community, there has been a handful of works that aim to combine stereo and DfD. Early approaches [16, 34] use a coarse estimation from DfD to reduce the search space of correspondence matching in stereo. Rajagopalan et al. [30] used a defocused stereo pair to recover depth and restore the all-focus image. Recently, Tao et al. [37] analyzed the variances of the epipolar plane image (EPI) to infer depth: the horizontal variance after vertical integration of the EPI encodes the defocus cue, while vertical variance represents the disparity cue. Both cues are then jointly optimized in a MRF framework. Takeda et al. [36] analyzed the relationship between point spread function and binocular disparity in the frequency domain, and jointly resolved the depth and deblurred the image. Wang et al. [38] presented a hybrid camera system that is composed of two calibrated auxiliary cameras and an uncalibrated main camera. The calibrated cameras were used to infer depth and the main camera provides DfD cues for boundary refinement. Our approach instead leverages the neural network to combine DfD and stereo estimations. To our knowledge, this is the first approach that employs deep learning for stereo and DfD fusion. 2. Training Data The key to any successful learning based depth inference scheme is a plausible training dataset. Numerous datasets have been proposed for stereo matching but very few are readily available for defocus based depth inference schemes. To address the issue, we set out to create a comprehensive DfD dataset. Our DfD dataset is based on FlyingThing3D [24], a synthetic dataset consisting of everyday objects randomly placed in the space. When generating the dataset, [24] separates the 3D models and textures into disjointed training and testing parts. In total there are 25,000 stereo images with ground truth disparities. In our dataset, we only

Figure 1. The top row shows the generated defocused image by using Virtual DSLR technique. The bottom row shows the ground truth color and depth images. We add Poisson noise to training data, a critical step for handling real scenes. The close-up views compare the defocused image with noise added and the original all-focus image. select stereo frames whose largest disparity is less than 100 pixels to avoid objects appearing in one image but not in the other. The synthesized color images in FlyingThings3D are allfocus images. To simulate defocused images, we adopt the Virtual DSLR approach from [40]. Virtual DSLR uses color and disparity image pair as input and outputs defocused image with quality comparable to those captured by expensive DSLR. The algorithm resembles refocusing technique in light field rendering without requiring the actual creation of the light field, thus reducing both memory and computational cost. Further, the Virtual DSLR takes special care of occlusion boundaries, to avoid color bleeding and discontinuity commonly observed on brute-force blur-based defocus synthesis. For the scope of this paper, we assume circular apertures, although more complex ones can easily be synthesized, e.g., for coded-aperture setups. To emulate different focus settings of the camera, we randomly set the focal plane, and select the size of the blur kernel in the range of 7 23 pixels. Finally, we add Poisson noise to both defocused image and the stereo pair to simulate the noise contained in real images. We d emphasize that the added noise is critical in real scene experiments, as will be discussed in 5.2. Our final training dataset contains 750 training samples and 160 testing samples, with each sample containing one stereo pair and the defocused image of the left view. The resolution of the generated images is 960 540, the same as the ones in FlyingThings3D. Figure 1 shows two samples of our training set. 3. DfD-Stereo Network Architecture Depth inference requires integration of both fine- and large-scale structures. For DfD and stereo, the depth cues could be distributed at various scales in an image. For instance, textureless background requires the understanding of a large region, while objects with complex shapes need attentive evaluation of fine details. To capture the contextual information across different scales, a number of recent approaches adopt multi-scale networks and the corresponding solutions have shown plausible results [7, 13]. In addition, recent studies [32] have shown that a deep network with small kernels is very effective in image recognition tasks. In comparison to large kernels, multiple layers of small kernels maintain a large receptive field while reducing the number of parameters to avoid overfitting. Therefore, a general principle in designing our network is a deep multi-scale architecture with small convolutional kernels. 3.1. Hourglass Network for DfD and Stereo Based on the observations above, we construct multi-scale networks that follow the hourglass (HG) architecture [25] for both DfD and stereo. Figure 2 illustrates the structure of our proposed network. HG network features a contractive part and an expanding part with skip layers between them. The contractive part is composed of convolution layers for feature extraction, and max pooling layers for aggregating high-level information over large areas. Specifically, we perform several rounds of max pooling to dramatically reduce the resolution, allowing smaller convolutional filters to be applied to extract features that span across the entire space of image. The expanding part is a mirrored architecture of the contracting part, with max pooling replaced by nearest neighbor upsampling layer. A skip layer that contains a residual module connects each pair of max pooling and upsampling layer so that the spatial information at each resolution will be preserved. Elementwise addition between the skip layer and the upsampled feature map follows to integrate the information across two adjacent resolutions. Both contractive and expanding part utilize large amount of residual modules [12]. Figure 2 (a) shows one HG structure.

M A E Label (a) M A E Label Input 1x1 conv, 128 Siamese Network HG Networks Intermediate Supervision After Each HG Deconv Network 3x3 conv, 128 1x1 conv, 256 (b) (c) Figure 2. (a) The hourglass network architecture consists of the max pooling layer (green), the nearest neighbor upsampling layer (pink), the residual module (blue), and convolution layer (yellow). The network includes intermediate supervision (red) to facilitate the training process. The loss function we use is mean absolute error (MAE). (b) The overall architecture of HG-DfD-Net and HG-Stereo-Net. The siamese network before the HG network aims to reduce the feature map size, while the deconvolution layers (gray) progressively recover the feature map to its original resolution. At each scale, the upsampled low resolution features are fused with high resolution features by using the concatenating layer (orange) (c) shows the detailed residual module. One pair of the contractive and expanding network can be viewed as one iteration of prediction. By stacking multiple HG networks together, we can further reevaluate and refine the initial prediction. In our experiment, we find a two-stack network is sufficient to provide satisfactory performance. Adding additional networks only marginally improves the results but at the expense of longer training time. Further, since our stacked HG network is very deep, we also insert auxiliary supervision after each HG network to facilitate the training process. Specifically, we first apply 1 1 convolution after each HG to generate an intermediate depth prediction. By comparing the prediction against the ground truth depth, we compute a loss. Finally, the intermediate prediction is remapped to the feature space by applying another 1 1 convolution, then added back to the features output from previous HG network. Our two-stack HG network has two intermediate loss, whose weight is equal to the weight of the final loss. Before the two-stack HG network, we add a siamese network, whose two network branches share the same architecture and weights. By using convolution layers that have a stride of 2, The siamese network serves to shrink the size of the feature map, thus reducing the memory usage and computational cost of the HG network. Compared with non-weight-sharing scheme, the siamese network requires fewer parameters and regularizes the network. After the HG network, we apply deconvolution layers to progressively recover the image to its original size. At each scale, the upsampled low resolution features are fused with high resolution features from the siamese network. This upsampling process with multi-scale guidance allows structures to be resolved at both fine- and large-scale. Note that based on our experiment, the downsample/upsample process largely facilitates the training and produces results that are very close

Defocused/ Focus Pair Siamese Network Conv Conv Hourglass Conv Conv Hourglass Concat Deconv Network Stereo Pair Siamese Network Figure 3. Architecture of HG-Fusion-Net. The convolution layers exchange information between networks at various stages, allowing the fusion of defocus and disparity cues. to those obtained from full resolution patches. Finally, the network produces pixel-wise disparity prediction at the end. The network architecture is shown in Figure 2 (b). For the details of layers, we use 2-D convolution of size 7 7 64 and 5 5 128 with stride 2 for the first two convolution layers in the siamese network. Each residual module contains three convolution layers as shown in Figure 2 (c). For the rest of convolution layers, they have kernel size of either 3 3 or 1 1. The input to the first hourglass is of quarter resolution and 256 channels while the output of the last hourglass is of the same shape. For the Deconv Network, the two 2-D deconvolution layers are of size 4 4 256 and 4 4 128 with stride 2. We use one such network for both DfD and stereo, which we call HG-DfD-Net and HG-Stereo-Net. The input of HG- DfD-Net is defocused/focus image pair of the left stereo view and the input of HG-Stereo-Net is stereo pair. 3.2. Network Fusion The most brute-force approach to integrate DfD and stereo is to directly concatenate the output disparity maps from the two branches and apply more convolutions. However, such an approach does not make use of the features readily presented in the branches and hence neglects cues for deriving the appropriate combination of the predicted maps. Consequently, such naïve approaches tend to average the results of two branches rather than making further improvement, as shown in Table 1. Instead, we propose HG-Fusion-Net to fuse DfD and stereo, as illustrated in figure 3. HG-Fusion-Net consists of two sub-networks, the HG-DfD-Net and HG-Stereo-Net. The inputs of HG-Fusion-Net are stereo pair plus the defocused image of the left stereo view, where the focused image of the left stereo view is fed into both the DfD and stereo sub-network. We set up extra connections between the two sub-networks. Each connection applies a 1 1 convolution on the features of one sub-network and adds to the other subnetwork. In doing so, the two sub-networks can exchange information at various stages, which is critical for different cues to interact with each other. The 1 1 convolution kernel serves as a transformation of feature space, consolidating new cues into the other branch. In our network, we set up pairs of interconnections at two spots, one at the beginning of each hourglass. At the cost of only four 1 1 convolutions, the interconnections largely proliferate the paths of the network. The HG-Fusion-Net can be regarded as an ensemble of original HG networks with different lengths that enables much stronger representation power. In addition, the fused network avoids solving the whole problem all at once, but first collaboratively solves the stereo and DfD sub-problems, then merges into one coherent solution. In addition to the above proposal, we also explore multiple variants of the HG-Fusion-Net. With no interconnection, the HG-Fusion-Net simply degrades to the brute-force approach. A compromise between our HG-Fusion-Net and the brute-force approach would be using only one pair of interconnections. We choose to keep the first pair, the one before the first hourglass, since it would enable the network to exchange information early. Apart from the number of interconnections, we also investigate the identity interconnections, which directly adds features to the other branch without going through 1 1 convolution. We present the quantitative results of all the models in Table 1. 4. Implementation Optimization All networks are trained in an end-to-end fashion. For the loss function, we use the mean absolute error (MAE) between ground truth disparity map and predicted disparity maps along with l 2 -norm regularization on parameters. We adopt MXNET [4] deep learning framework to implement and train our models. Our implementation applies batch normalization [14] after each convolution layer, and use PReLU layer [11] to add nonlinearity to the network while avoiding the dead ReLU. We also use the technique from [11] to initialize the weights. For the network solver we choose the Adam optimizer [15] and set the initial learning rate = 0.001, weight decay = 0.002, β1 = 0.9, β2 = 0.999. We train and test all the models on a NVIDIA Tesla K80 graphic card. Data Preparation and Augmentation To prepare the data,

All-Focus Image (L) HG-DfD-Net HG-Stereo-Net HG-Fusion-Net Ground Truth (a) Side View Front View HG-DfD-Net HG-Stereo-Net HG-Fusion-Net Ground Truth (b) Figure 4. Results of HG-DfD-Net, HG-Stereo-Net and HG-Fusion-Net on (a) our dataset (b) staircase scene textured with horizontal stripes. HG-Fusion-Net produces smooth depth at flat regions while maintaining sharp depth boundaries. Best viewed in the electronic version by zooming in. we first stack the stereo/defocus pair along the channel s direction, then extract patches from the stacked image with a stride of 64 to increase the number of training samples. Recall that the HG network contains multiple max pooling layers for downsampling, the patch needs to be cropped to the nearest number that is multiple of 64 for both height and width. In the training phase, we use patches of size 512 256 as input. The large patch contains enough contextual information to recover depth from both defocus and stereo. To increase the generalization of the network, we also augment the data by flipping the patches horizontally and vertically. We perform the data augmentation on the fly at almost no additional cost. 5. Experiments 5.1. Synthetic Data We train HG-DfD-Net, HG-Stereo-Net and HG-FusionNet separately, and then conduct experiments on test samples from synthetic data. Figure 4(a) compares the results of these three networks. We observe that results from HG-DfD-Net show clearer depth edge, but also exhibit noise on flat regions. On the contrary, HG-Stereo-Net provides smooth depth. However, there is depth bleeding across boundaries, especially when there are holes, such as the tire of the motorcycle on the first row. We suspect that the depth bleeding is due to occlusion, by which the DfD is less affected. Finally, HG-Fusion-Net finds the optimal combination of the two, producing smooth depth while keeping sharp depth boundaries. Table 1 also quantitatively compares the performance of different models on our synthetic dataset. Results from Table 1 confirm that HG-Fusion-Net achieves the best result for almost all metrics, with notable margin ahead of using stereo or defocus cues alone. The brute-force fusion approach with no interconnection only averages results from HG-DfD-Net and HG-Stereo-Net, thus is even worse than HG-Stereo-Net alone. The network with fewer or identity interconnection performs slightly worse than HG-Fusion-Net, but still a lot better than the network with no interconnection. This demonstrates that interconnections can efficiently

> 1 px > 3 px > 5 px MAE (px) Time (s) HG-DfD-Net 70.07% 38.60% 20.38% 3.26 0.24 HG-Stereo-Net 28.10% 6.12% 2.91% 1.05 0.24 HG-Fusion-Net 20.79% 5.50% 2.54% 0.87 0.383 No Interconnection 45.46% 10.89% 5.08% 1.57 0.379 Less Interconnection 21.85% 5.23% 2.55% 0.91 0.382 Identity Interconnection 21.37% 6.00% 2.96% 0.94 0.382 MC-CNN-fast [43] 15.38% 10.84% 9.25% 2.76 1.94 MC-CNN-acrt [43] 13.95% 9.53% 8.11% 2.35 59.77 Table 1. Quantitative results on synthetic data. Upper part compares results from different input combinations: defocus pair, stereo pair and stereo pair + defocused image. Middle part compares various fusion scheme, mainly differentiating by the number and type of interconnection: No interconnection is the brute-force approach that only concatenates feature maps after the HG network, before the deconvolution layers. Less Interconnection only uses one interconnection before the first hourglass; Identity Interconnection directly adds features to the other branch, without applying the 1 1 convolution. Lower part shows results of [43]. The metrics > 1 px, > 3 px, > 5 px represent the percentage of pixels whose absolute disparity error is larger than 1, 3, 5 pixels respectively. MAE measures the mean absolute error of disparity map. broadcast information across branches and largely facilitate mutual optimization. We also compare our models with the two stereo matching approaches from [43] in Table 1. These approaches utilize CNN to compute the matching costs and use them to carry out cost aggregation and semiglobal matching, followed by post-processing steps. While the two approaches have fewer pixels with error larger than 1 pixel, they yield more large-error pixels and thus have worse overall performance. In addition, their running time is much longer than our models. We also conduct another experiment on a scene with a staircase textured by horizontal stripes, as illustrated in figure 4(b). The scene is rendered from the front view, making it extremely challenging for stereo since all the edges are parallel to the epipolar line. On the contrary, DfD will be able to extract the depth due to its 2D aperture. Figure 4(b) shows the resultant depths enclosed in the red box of the front view, proving the effectiveness of our learning-based DfD on such difficult scene. Note that the inferred depth is not perfect. This is mainly due to the fact that our training data lacks objects with stripe texture. We can improve the result by adding similar textures to the training set. 5.2. Real Scene To conduct experiments on the real scene, we use light field (LF) camera to capture the LF and generate the defocused image. While it is possible to generate the defocused image using conventional cameras by changing the aperture size, we find it difficult to accurately match the light throughput, resulting in different brightness between the all-focused and defocused image. The results using conventional cameras are included in supplementary material. LF camera captures a rich set of rays to describe the visual appearance of the scene. In free space, LF is commonly represented by two-plane parameterizations L(u, v, s, t), where st is the camera plane and uv is the image plane [19]. To conduct digital refocusing, we can move the synthetic image plane that leads to the following photography equation [26]: E(s, t) = L(u, v, u + s u α, v + t v )dudv (1) α By varying α, we can refocus the image at different depth. Note that by fixing st, we obtain the sub-aperture image L (s t )(u, v) that is amount to the image captured using a sub-region of the main lens aperture. Therefore, Eqn. 1 corresponds to shift-and-add the sub-aperture images [26]. In our experiment, we use Lytro Illum camera as our capturing device. We first mount the camera on a translation stage and move the LF camera horizontally to capture two LFs. Then we extract the sub-aperture images from each LF using Light Field Toolbox [6]. The two central sub-aperture images are used to form a stereo pair. We also use the central sub-aperture image in the left view as the all-focused image due to its small aperture size. Finally, we apply the shift-and-add algorithm to generate the defocused image. Both the defocused and sub-aperture images have the size of 625 433. We have conducted tests on both outdoor and indoor scenes. In Fig.5, we compare the performance of our different models. In general, both HG-DfD-Net and HG-Stereo- Net preserve depth edges well, but results from HG-DfD-Net are noisier. In addition, the result of HG-DfD-Net is inaccurate at positions distant from camera because defocus blurs vary little at large depth. HG-Fusion-Net produces the best results with smooth depth and sharp depth boundaries. We have also trained HG-Fusion-Net on a clean dataset without Poisson noise, and show the results in the last column of

All-Focus Image (L) HG-DfD-Net HG-Stereo-Net HG-Fusion-Net HG-Fusion-Net (Clean) Figure 5. Comparisons of real scene results from HG-DfD-Net, HG-Stereo-Net and HG-Fusion-Net. The last column shows the results from HG-Fusion-Net trained by the clean dataset without Poisson noise. Best viewed in color. All-Focus Image (L) Ours MC-CNN-fast [43] MC-CNN-acrt [43] Figure 6. Comparisons of real scene results with [43]. Best viewed in color. Fig.5. The inferred depths exhibit severe noise pattern on real data, confirming the necessity to add noise to the dataset for simulating real images. In Fig.6, we compare our approach with stereo matching methods from [43]. The plant and the toy tower in the first two rows present challenges for stereo matching due to the heavy occlusion. By incorporating both DfD and stereo, our approach manages to recover the fine structure of leaves and segments as shown in the zoomed regions while methods from [43] either over-smooth or wrongly predict at these positions. The third row further demonstrates the efficacy of our approach on texture-less or striped regions. 6. Conclusion We have presented a learning based solution for a hybrid DfD and stereo depth sensing scheme. We have adopted the hourglass network architecture to separately extract depth from defocus and stereo. We have then studied and explored multiple neural network architectures for linking both networks to improve depth inference. Comprehensive experiments show that our proposed approach preserves the strength of DfD and stereo while effectively suppressing their weaknesses. In addition, we have created a large synthetic dataset for our setup that includes image triplets of a stereo pair and a defocused image along with the corresponding ground truth disparity. Our immediate future work is to explore different DfD inputs and their interaction with stereo. For instance, instead of using a single defocused image, we can vary the aperture size to produce a stack of images where objects at the same depth exhibit different blur profiles. Learning-based approaches can be directly applied to the profile for depth inference or can be combined with our current framework for conducting hybrid depth inference. We have presented one DfD-Stereo setup. Another minimal design was shown in [36], where a stereo pair with different focus distance is used as input. In the future, we will study the cons and

pros of different hybrid DfD-stereo setups and tailor suitable learning-based solutions for fully exploiting the advantages of such setups. References [1] E. Alexander, Q. Guo, S. J. Koppal, S. J. Gortler, and T. E. Zickler. Focal flow: Measuring distance and velocity with defocus and differential motion. In ECCV, pages 667 682, 2016. [2] V. M. Bove. Entropy-based depth from focus. Journal of the Optical Society of America, 10(10):561 566, 1993. [3] M. Z. Brown, D. Burschka, and G. D. Hager. Advances in computational stereo. TPAMI, 25(8):993 1008, 2003. [4] T. Chen, M. Li, Y. Li, M. Lin, N. Wang, M. Wang, T. Xiao, B. Xu, C. Zhang, and Z. Zhang. Mxnet: A flexible and efficient machine learning library for heterogeneous distributed systems. CoRR, abs/1512.01274, 2015. [5] Z. Chen, X. Sun, L. Wang, Y. Yu, and C. Huang. A deep visual correspondence embedding model for stereo matching costs. In ICCV, pages 972 980, 2015. [6] D. Dansereau, O. Pizarro, and S. Williams. Decoding, calibration and rectification for lenselet-based plenoptic cameras. In CVPR, pages 1027 1034, 2013. [7] D. Eigen, C. Puhrsch, and R. Fergus. Depth map prediction from a single image using a multi-scale deep network. In NIPS, 2014. [8] J. Ens and P. Lawrence. A matrix based method for determining depth from focus. In CVPR, pages 600 606, 1991. [9] P. Favaro, S. Soatto, M. Burger, and S. Osher. Shape from defocus via diffusion. TPAMI, 30:518 531, 2008. [10] X. Han, T. Leung, Y. Jia, R. Sukthankar, and A. C. Berg. Matchnet: Unifying feature and metric learning for patchbased matching. In CVPR, pages 3279 3286, 2015. [11] K. He, X. Zhang, S. Ren, and J. Sun. Delving deep into rectifiers: Surpassing human-level performance on imagenet classification. ICCV, pages 1026 1034, 2015. [12] K. He, X. Zhang, S. Ren, and J. Sun. Deep residual learning for image recognition. In CVPR, pages 770 778, 2016. [13] T.-W. Hui, C. C. Loy, and X. Tang. Depth map superresolution by deep multi-scale guidance. In ECCV, 2016. [14] S. Ioffe and C. Szegedy. Batch normalization: Accelerating deep network training by reducing internal covariate shift. In ICML, pages 448 456, 2015. [15] D. P. Kingma and J. Ba. Adam: A method for stochastic optimization. ICLR, 2015. [16] W. N. Klarquist, W. S. Geisler, and A. C. Bovik. Maximumlikelihood depth-from-defocus for active vision. In IROS, 1995. [17] P. Knöbelreiter, C. Reinbacher, A. Shekhovtsov, and T. Pock. End-to-end training of hybrid cnn-crf models for stereo. CVPR, pages 1456 1465, 2017. [18] A. Levin, R. Fergus, F. Durand, and W. T. Freeman. Image and depth from a conventional camera with a coded aperture. ACM Trans. Graph., 26(3), 2007. [19] M. Levoy and P. Hanrahan. Light field rendering. In SIG- GRAPH, 1996. [20] N. Li, B. Sun, and J. Yu. A weighted sparse coding framework for saliency detection. In CVPR, pages 5216 5223, 2015. [21] N. Li, J. Ye, Y. Ji, H. Ling, and J. Yu. Saliency detection on light field. In CVPR, pages 2806 2813, 2014. [22] Z. Liu, Z. Li, J. Zhang, and L. Liu. Euclidean and hamming embedding for image patch description with convolutional networks. In CVPR Workshops, pages 72 78, 2016. [23] W. Luo, A. G. Schwing, and R. Urtasun. Efficient deep learning for stereo matching. In TPAMI, pages 5695 5703, 2016. [24] N. Mayer, E. Ilg, P. Häusser, P. Fischer, D. Cremers, A. Dosovitskiy, and T. Brox. A large dataset to train convolutional networks for disparity, optical flow, and scene flow estimation. CVPR, pages 4040 4048, 2016. [25] A. Newell, K. Yang, and J. Deng. Stacked hourglass networks for human pose estimation. In ECCV, pages 483 499. Springer, 2016. [26] R. Ng, M. Levoy, M. Bredif, G. Duval, M. Horowitz, and P. Hanrahan. Light field photography with a hand-held plenoptic camera. Stanford University Computer Science Tech Report, 2:1 11, 2005. [27] H. Park and K. M. Lee. Look wider to match image patches with convolutional neural networks. IEEE Signal Processing Letters, 2016. [28] A. P. Pentland. A new sense for depth of field. TPAMI, pages 523 531, 1987. [29] A. N. Rajagopalan and S. Chaudhuri. Optimal selection of camera parameters for recovery of depth from defocused images. In CVPR, pages 219 224, 1997. [30] A. N. Rajagopalan, S. Chaudhuri, and U. Mudenagudi. Depth estimation and image restoration using defocused stereo pairs. TPAMI, 26(11):1521 1525, 2004. [31] D. Scharstein and R. Szeliski. A taxonomy and evaluation of dense two-frame stereo correspondence algorithms. IJCV, 47:7 42, 2002. [32] K. Simonyan and A. Zisserman. Very deep convolutional networks for large-scale image recognition. CoRR, abs/1409.1556, 2014. [33] M. Subbarao and G. Surya. Depth from defocus: A spatial domain approach. IJCV, 13:271 294, 1994. [34] M. Subbarao, T. Yuan, and J. Tyan. Integration of defocus and focus analysis with stereo for 3d shape recovery. In Proc. SPIE, volume 3204, pages 11 23, 1997. [35] G. Surya and M. Subbarao. Depth from defocus by changing camera aperture: a spatial domain approach. CVPR, pages 61 67, 1993. [36] Y. Takeda, S. Hiura, and K. Sato. Fusing depth from defocus and stereo with coded apertures. In CVPR, pages 209 216, 2013. [37] M. W. Tao, S. Hadap, J. Malik, and R. Ramamoorthi. Depth from combining defocus and correspondence using light-field cameras. ICCV, pages 673 680, 2013. [38] T. C. Wang, M. Srikanth, and R. Ramamoorthi. Depth from semi-calibrated stereo and defocus. In CVPR, pages 3717 3726, 2016. [39] M. Watanabe and S. K. Nayar. Rational filters for passive depth from defocus. IJCV, 27:203 225, 1998.

[40] Y. Yang, H. Lin, Z. Yu, S. Paris, and J. Yu. Virtual dslr: High quality dynamic depth-of-field synthesis on mobile platforms. In Digital Photography and Mobile Imaging, 2016. [41] S. Zagoruyko and N. Komodakis. Learning to compare image patches via convolutional neural networks. CVPR, pages 4353 4361, 2015. [42] J. Zbontar and Y. LeCun. Computing the stereo matching cost with a convolutional neural network. CVPR, pages 1592 1599, 2015. [43] J. Zbontar and Y. LeCun. Stereo matching by training a convolutional neural network to compare image patches. Journal of Machine Learning Research, 17(1-32):2, 2016. [44] C. Zhou, S. Lin, and S. Nayar. Coded aperture pairs for depth from defocus. In ICCV, pages 325 332, 2010.

A. Results On Data Captured By Conventional Cameras We have captured real scene image triplets using a pair of DSLR cameras. We capture the defocused left image by changing the aperture of cameras. Figure 7 shows one example of image triplet. Notice that the brightness of all-focus and defocused left image is inconsistent near image borders. We also compare our results with [43] on our data captured with DSLR cameras, as shown in Figure 8. All-Focus Image (L) Defocused Image (L) All-Focus Image (R) Figure 7. An Example of our captured data. All-Focus Image (L) HG-DfD-Net HG-Stereo-Net HG-Fusion-Net MC-CNN-fast [43] MC-CNN-acrt [43] Figure 8. Results on data captured by DSLR cameras. The first three columns are results of our different models while the last two columns are results of [43].