Skip to content

skth5199/depth_gen

Repository files navigation

Three Depth estimation end to end approaches focusing on complex situations while reducing computational complexity

Three monocular depth estimation approaches were developed:

  1. Pix2pix model
  2. U-net with DenseNet encoder
  3. U-net with MobileNetv2 encoder

Datasets used for training and testing:

  1. KITTI
  2. NYU

A sample of the input and the resulting depth maps are shown below:

Each row consists of the input image, ground truth depth map, output by pix2pix model, output by U-net with MobileNetv2 and output by U-net with Densenet encoder, respectively from right to left.

To generate depth maps for a custom image, simply download and run as follows:
$ python gui.py

The architectures of these approaches is illustrated in the dissertation document.

The engineered metric

A new metric was engineered for the fair and more apposite comparison of the approaches, which factored in the computational complexity of the approaches. It consists of two components. The computational complexity (CC) is calculated as follows:

where n is the total number of approaches being considered.

The final value of the metric is calculated as follows:

Conclusion

It was determined that the Pix2pix model performed the best with an SSIM and Engineered metric scores of over 0.95. The testing process involved both, indoor and outdoor scenes. The U-nets with Densenet and MobileNetv2 encoders scored reltively less on these metrics. However, the generated depth maps were further evaluated and explained through inspection to even out the differences caused by various factors such as colour encodings, between the ground truth maps and the depth maps generated by the approaches. Here, it was determined that the U-net with MobileNetv2 also generated accurate depth maps. Its score was low solely due to the aforementioned differences. Finally, a Graphical user interface was developed to facilitate the hybridised and appropriate use of the engineered models.

Releases

No releases published

Packages

No packages published

Languages