Skip to content

sophiajwagner/depth-image-enhancement-for-UAV-exploration

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Depth image enhancement for UAV exploration

A project within the class "ECE 740 - Computer and Robot Vision" at University of Alberta.
Shane Allan, Sophia Wagner, Almas Sahar

In this work, an enhanced depth estimation methodology is proposed using a low-cost stereo-vision camera that can deliver quality depth images in low-light environments suitable for real-time implementation. To address this problem, a convolutional neural network (CNN) is recommended to enhance the raw stereo red, green, and blue (RGB) and depth images for use on a prototype unmanned aerial vehicle (UAV). Our contributions are the following:

  • Development of different CNN models in order to enhance depth from low-light depth and stereo images.
  • Generation of a dataset with varying quality and lighting conditions consisting of depth and stereo images.
  • Running the CNN models on the onboard computing module of the UAV.

The generated dataset can be found here


Resources

Software:

  • PyTorch 1.10.0
  • Python 2.7, 3.6
  • C++
  • ROS Melodic
  • Ubuntu 18.04

Hardware:

  • DJI F450 Flamewheel Frame
  • Garmin ToF Laser Rangefinder
  • Pixhawk 5x Flight Controller
  • Sereolabs Zed 2 Camera
  • Jetson Xavier NX

Results

Architecture Results
Architecture Results

About

Depth image enhancement for autonomous UAV exploration

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published