ChristianOrr
commited on
Commit
•
3fcc36a
1
Parent(s):
5946209
Updated readme
Browse files
README.md
CHANGED
@@ -15,14 +15,14 @@ datasets:
|
|
15 |
|
16 |
MADNet is a deep stereo depth estimation model. Its key defining features are:
|
17 |
1. It has a light-weight architecture which means it has low latency.
|
18 |
-
2. It supports self-supervised training, so it can be
|
19 |
-
3.
|
20 |
|
21 |
The MADNet weights in this repository were trained using a Tensorflow 2 / Keras implementation of the original code. The model was created using the Keras Functional API, which enables the following features:
|
22 |
-
1. Good optimization
|
23 |
2. High level Keras methods (.fit, .predict and .evaluate).
|
24 |
-
3.
|
25 |
-
4. Decent support from external
|
26 |
5. Callbacks.
|
27 |
|
28 |
The weights provided were either trained on the 2012 / 2015 kitti stereo dataset or flyingthings-3d dataset. The weights of the pretrained models from the original paper (tf1_conversion_kitti.h5 and tf1_conversion_synthetic.h5) are provided in tensorflow 2 format. The TF1 weights help speed up fine-tuning, but its recommended to use either synthetic.h5 (trained on flyingthings-3d) or kitti.h5 (trained on 2012 and 2015 kitti stereo datasets).
|
|
|
15 |
|
16 |
MADNet is a deep stereo depth estimation model. Its key defining features are:
|
17 |
1. It has a light-weight architecture which means it has low latency.
|
18 |
+
2. It supports self-supervised training, so it can be conveniently adapted in the field with no training data.
|
19 |
+
3. It's a stereo depth model, which means it's capable of high accuracy.
|
20 |
|
21 |
The MADNet weights in this repository were trained using a Tensorflow 2 / Keras implementation of the original code. The model was created using the Keras Functional API, which enables the following features:
|
22 |
+
1. Good optimization.
|
23 |
2. High level Keras methods (.fit, .predict and .evaluate).
|
24 |
+
3. Little boilerplate code.
|
25 |
+
4. Decent support from external packages (like Weights and Biases).
|
26 |
5. Callbacks.
|
27 |
|
28 |
The weights provided were either trained on the 2012 / 2015 kitti stereo dataset or flyingthings-3d dataset. The weights of the pretrained models from the original paper (tf1_conversion_kitti.h5 and tf1_conversion_synthetic.h5) are provided in tensorflow 2 format. The TF1 weights help speed up fine-tuning, but its recommended to use either synthetic.h5 (trained on flyingthings-3d) or kitti.h5 (trained on 2012 and 2015 kitti stereo datasets).
|