Wolfram Research

Single-Image Depth Perception Net Trained on Depth in the Wild Data

Estimate the depth map of an image

Released in 2016, this neural net was trained to predict the relative depth map from a single image using a novel technique based on sparse ordinal annotations. Each training example only needs to be annotated with a pair of points and its relative distance to the camera. After training, the net is able to reconstruct the full depth map. Its architecture is based on the "hourglass" design.

Number of layers: 501 | Parameter count: 5,385,185 | Trained size: 23 MB

Training Set Information

Performance

Examples

Resource retrieval

Get the pre-trained net:

In[1]:=
NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"]
Out[1]=

Basic usage

Obtain the depth map of an image:

In[2]:=
CloudGet["https://www.wolframcloud.com/objects/b14bc770-5d09-40b1-a4fb-19c9de229548"] (* Evaluate this cell to copy the example input from a cloud object *)
Out[2]=

Show the depth map:

In[3]:=
ImageAdjust[Image[depthMap]]
Out[3]=

Visualize a 3D model

Get an image:

In[4]:=
CloudGet["https://www.wolframcloud.com/objects/d1cc4b1e-f960-462a-9aa7-4de32ef7a9d3"] (* Evaluate this cell to copy the example input from a cloud object *)

Obtain the depth map:

In[6]:=
depthMap = NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"][img]
Out[6]=

Visualize a 3D model using the depth map:

In[7]:=
ListPlot3D[-depthMap, PlotStyle -> Texture[ImageReflect[img]], PlotTheme -> {"Minimal", "NoAxes"}, ViewPoint -> {0, 0.76, 1.84}]
Out[7]=

Adapt to any size

The recommended way to deal with image sizes and aspect ratios is to resample the depth map after the net evaluation. Get an image:

In[8]:=
CloudGet["https://www.wolframcloud.com/objects/332306db-8643-48bc-a7f9-9829dde17ff9"] (* Evaluate this cell to copy the example input from a cloud object *)

Obtain the dimensions of the image:

In[9]:=
ImageDimensions[img]
Out[9]=

Obtain the depth map

In[10]:=
depthMap = NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"][img]
Out[10]=

Resample the depth map and visualize it:

In[11]:=
ImageAdjust[ Image[ArrayResample[depthMap, Reverse[ImageDimensions[img]]]]]
Out[11]=

Net information

Inspect the number of parameters of all arrays in the net:

In[12]:=
NetInformation[ NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"], "ArraysElementCounts"]
Out[12]=

Obtain the total number of parameters:

In[13]:=
NetInformation[ NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"], "ArraysTotalElementCount"]
Out[13]=

Obtain the layer type counts:

In[14]:=
NetInformation[ NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"], "LayerTypeCounts"]
Out[14]=

Display the summary graphic:

In[15]:=
NetInformation[ NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"], "SummaryGraphic"]
Out[15]=

Export to MXNet

Export the net into a format that can be opened in MXNet:

In[16]:=
jsonPath = Export[FileNameJoin[{$TemporaryDirectory, "net.json"}], NetModel["Single-Image Depth Perception Net Trained on Depth in the \ Wild Data"], "MXNet"]
Out[16]=

Export also creates a net.params file containing parameters:

In[17]:=
paramPath = FileNameJoin[{DirectoryName[jsonPath], "net.params"}]
Out[17]=

Get the size of the parameter file:

In[18]:=
FileByteCount[paramPath]
Out[18]=

The size is similar to the byte count of the resource object:

In[19]:=
ResourceObject[ "Single-Image Depth Perception Net Trained on Depth in the Wild \ Data"]["ByteCount"]
Out[19]=

Requirements

Wolfram Language 11.2 (September 2017) or above

Resource History

  • Date Created: 13 October 2017
  • Latest Update: 21 June 2018

Reference