Skip to content

CNN that uses UNet architecture to predict the thickness of objects in an image

Notifications You must be signed in to change notification settings

kane1204/Object-Thickness-Estimation-Network

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

59 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Object-Thickness-Estimation-Network

CNN that uses UNet architecture to predict the thickness of objects in an image. This profiles the deep learning network for my masters project part 2.

Abstract

In this project, we investigate a novel task related to understanding 3D scenes from single images, called object thickness estimation. Our primary objective was to create a neural network capable of predicting the thickness of an object. Due to the lack of data for this task, we had to develop a repeatable method for creating bespoke synthetic datasets for training, validation, and testing. This provides insights into the challenges associated with generating synthetic data. We trained an adapted U-Net model to predict the thickness of a single object in an image and evaluated its performance against a test set. We also conducted multiple experiments, one of which offered more insight into the dependence on context of objects when performing this task. Furthermore, we compared the performance of human estimators to the fully trained U-Net model by quizzing them on relative thickness questions. The model showed promise quantitatively, performing better in every way compared to the baseline and exhibiting strong performance against human estimators.

About

CNN that uses UNet architecture to predict the thickness of objects in an image

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published