diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..ec4a626 --- /dev/null +++ b/.gitattributes @@ -0,0 +1 @@ +*.pth filter=lfs diff=lfs merge=lfs -text diff --git a/README.md b/README.md index 79384e7..d24f5f6 100644 --- a/README.md +++ b/README.md @@ -9,8 +9,10 @@ This is the corresponding code to the above paper ("Self-supervised Learning of Applications") which is published at the International Conference on Robotics and Automation (ICRA) 2021. The code is provided by the [Robotics Systems Lab](https://rsl.ethz.ch/) at ETH Zurich, Switzerland. -**Authors:** [Julian Nubert](https://juliannubert.com/) ([julian.nubert@mavt.ethz.ch](mailto:julian.nubert@mavt.ethz.ch?subject=[GitHub])) -, [Shehryar Khattak](https://www.linkedin.com/in/shehryar-khattak/), [Marco Hutter](https://rsl.ethz.ch/the-lab/people/person-detail.MTIxOTEx.TGlzdC8yNDQxLC0xNDI1MTk1NzM1.html) +** +Authors:** [Julian Nubert](https://juliannubert.com/) ([julian.nubert@mavt.ethz.ch](mailto:julian.nubert@mavt.ethz.ch?subject=[GitHub])) +, [Shehryar Khattak](https://www.linkedin.com/in/shehryar-khattak/) +, [Marco Hutter](https://rsl.ethz.ch/the-lab/people/person-detail.MTIxOTEx.TGlzdC8yNDQxLC0xNDI1MTk1NzM1.html) ![title_img](images/title_img.png) @@ -153,6 +155,8 @@ run_testing.py --checkpoint The checkpoint can be found in MLFlow after training. It runs testing for the dataset specified in [./config/deployment_options.yaml](./config/deployment_options.yaml). +We provide an exemplary trained model in [./checkpoints/kitti_example.pth](./checkpoints/kitti_example.pth). + ### ROS-Node This ROS-node takes the pretrained model at location `````` and performs inference; i.e. it predicts and diff --git a/checkpoints/README.md b/checkpoints/README.md new file mode 100644 index 0000000..791fbf9 --- /dev/null +++ b/checkpoints/README.md @@ -0,0 +1,14 @@ +# Example network trained on the KITTI dataset + +We provide an example network in ```kitti_example.pth```. Note that the checkpoint also includes the configuration of +the trained model. When loading the model, the corresponding configuration is set automatically. + +The plots generated by the testing code for this model can be seen in the following: + +| Sequence 00 | Sequence 02 | +| :---: | :---: | +| ![Dataset 00](kitti_00_2d.png) | ![Dataset 02](kitti_02_2d.png) | +| Sequence 05 | Sequence 07 | +| ![Dataset 05](kitti_05_2d.png) | ![Dataset 08](kitti_07_2d.png) | +| Sequence 08 | Sequence 10 | +| ![Dataset 08](kitti_08_2d.png) | ![Dataset 10](kitti_10_2d.png) | diff --git a/checkpoints/kitti_00_2d.png b/checkpoints/kitti_00_2d.png new file mode 100644 index 0000000..e57b8f3 Binary files /dev/null and b/checkpoints/kitti_00_2d.png differ diff --git a/checkpoints/kitti_02_2d.png b/checkpoints/kitti_02_2d.png new file mode 100644 index 0000000..5e932d6 Binary files /dev/null and b/checkpoints/kitti_02_2d.png differ diff --git a/checkpoints/kitti_05_2d.png b/checkpoints/kitti_05_2d.png new file mode 100644 index 0000000..9d883ad Binary files /dev/null and b/checkpoints/kitti_05_2d.png differ diff --git a/checkpoints/kitti_07_2d.png b/checkpoints/kitti_07_2d.png new file mode 100644 index 0000000..c869ee4 Binary files /dev/null and b/checkpoints/kitti_07_2d.png differ diff --git a/checkpoints/kitti_08_2d.png b/checkpoints/kitti_08_2d.png new file mode 100644 index 0000000..4a8ce6f Binary files /dev/null and b/checkpoints/kitti_08_2d.png differ diff --git a/checkpoints/kitti_10_2d.png b/checkpoints/kitti_10_2d.png new file mode 100644 index 0000000..229de45 Binary files /dev/null and b/checkpoints/kitti_10_2d.png differ diff --git a/checkpoints/kitti_example.pth b/checkpoints/kitti_example.pth new file mode 100644 index 0000000..717a2bd --- /dev/null +++ b/checkpoints/kitti_example.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:949dc2080b388be54e21abc3dce6a236931ef9df1a17af767002bd6ea03aec34 +size 142547609