vumichien commited on
Commit
fa9bc19
1 Parent(s): 4585388

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-to-image
4
+ library_name: keras
5
+ ---
6
+ ## Model description
7
+ Depth estimation is a crucial step towards inferring scene geometry from 2D images. The goal in monocular depth estimation is to predict the depth value of each pixel or inferring depth information, given only a single RGB image as input.
8
+
9
+ ## Dataset
10
+ [NYU Depth Dataset V2](https://cs.nyu.edu/~silberman/datasets/nyu_depth_v2.html) is comprised of video sequences from a variety of indoor scenes as recorded by both the RGB and Depth cameras from the Microsoft Kinect.
11
+
12
+ ## Training procedure
13
+
14
+ ### Training hyperparameters
15
+ **Model architecture**:
16
+ - UNet with a pretrained DenseNet 201 backbone.
17
+
18
+ The following hyperparameters were used during training:
19
+ - learning_rate: 1e-04
20
+ - train_batch_size: 16
21
+ - seed: 42
22
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
23
+ - lr_scheduler_type: ReduceLROnPlateau
24
+ - num_epochs: 10
25
+
26
+ ### Training results
27
+
28
+ | Epoch | Training loss | Validation Loss | Learning rate |
29
+ |:------:|:-------------:|:---------------:|:-------------:|
30
+ | 1 | 0.1333 | 0.1315 | 1e-04 |
31
+ | 2 | 0.0948 | 0.1232 | 1e-04 |
32
+ | 3 | 0.0834 | 0.1220 | 1e-04 |
33
+ | 4 | 0.0775 | 0.1213 | 1e-04 |
34
+ | 5 | 0.0736 | 0.1196 | 1e-04 |
35
+ | 6 | 0.0707 | 0.1205 | 1e-04 |
36
+ | 7 | 0.0687 | 0.1190 | 1e-04 |
37
+ | 8 | 0.0667 | 0.1177 | 1e-04 |
38
+ | 9 | 0.0654 | 0.1177 | 1e-04 |
39
+ | 10 | 0.0635 | 0.1182 | 9e-05 |
40
+ ### View Model Demo
41
+ ![Model Demo](./demo.png)
42
+
43
+ ### View Model Plot
44
+ ![Model Image](./model.png)