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).
|