Skip to content

A Pytorch implement of 《Deeper Depth Prediction with Fully Convolutional Residual Networks》

Notifications You must be signed in to change notification settings

mhmedwagih1324/Depth-Estimation-Model

 
 

Repository files navigation

Depth-Estimation-Model

This model was created to predict the depth of a single RGB image of food after transfering a depth model that had been used to predict the depths of indoor scenes, to predict the depth of food images.

Idea

Our idea is to use our small dataset to create a model that predicts the depth of a food image starting from the pretrained Tensorflow weights.

Content

Some codes of this repository are already implemented in this repositoy

Method

To start training, please follow the following steps:

  1. open the depth_train_code.ipynb on Google colab
  2. run all the cells till the cell that starts training, these cells should import the required libraries, clone this repository and download the required model and dataset.
  3. run the cell that starts training the model.
  4. run the cell that starts testing the model.
  5. the output images of training and testing should appear in the same directory that contains this cloned repository

Results

Our training & testing results are in the excel sheet Depth_training_results.xlsx Also some of the output images of train and tests appear in the folder Training out images train results

About

A Pytorch implement of 《Deeper Depth Prediction with Fully Convolutional Residual Networks》

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 91.0%
  • Python 9.0%