vumichien commited on
Commit
4621f87
2 Parent(s): 0629d73 9e9f78e

Merge remote-tracking branch 'origin/main'

Browse files
Files changed (1) hide show
  1. README.md +56 -0
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-segmentation
4
+ library_name: keras
5
+ ---
6
+ ## Model description
7
+ Full credits go to: [Vu Minh Chien](https://www.linkedin.com/in/vumichien/)
8
+
9
+ 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.
10
+
11
+ ## Dataset
12
+ [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.
13
+
14
+ ## Training procedure
15
+
16
+ ### Training hyperparameters
17
+ **Model architecture**:
18
+ - UNet with a pretrained DenseNet 201 backbone.
19
+
20
+ The following hyperparameters were used during training:
21
+ - learning_rate: 1e-04
22
+ - train_batch_size: 16
23
+ - seed: 42
24
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
25
+ - lr_scheduler_type: ReduceLROnPlateau
26
+ - num_epochs: 10
27
+
28
+ ### Training results
29
+
30
+ | Epoch | Training loss | Validation Loss | Learning rate |
31
+ |:------:|:-------------:|:---------------:|:-------------:|
32
+ | 1 | 0.1333 | 0.1315 | 1e-04 |
33
+ | 2 | 0.0948 | 0.1232 | 1e-04 |
34
+ | 3 | 0.0834 | 0.1220 | 1e-04 |
35
+ | 4 | 0.0775 | 0.1213 | 1e-04 |
36
+ | 5 | 0.0736 | 0.1196 | 1e-04 |
37
+ | 6 | 0.0707 | 0.1205 | 1e-04 |
38
+ | 7 | 0.0687 | 0.1190 | 1e-04 |
39
+ | 8 | 0.0667 | 0.1177 | 1e-04 |
40
+ | 9 | 0.0654 | 0.1177 | 1e-04 |
41
+ | 10 | 0.0635 | 0.1182 | 9e-05 |
42
+
43
+
44
+
45
+ ### View Model Demo
46
+
47
+ ![Model Demo](./demo.png)
48
+
49
+
50
+ <details>
51
+
52
+ <summary> View Model Plot </summary>
53
+
54
+ ![Model Image](./model.png)
55
+
56
+ </details>