change some dir name

RandyWangRaining · Jun 24, 2024 · 59b6c01 · 59b6c01
1 parent 85c8d83
commit 59b6c01
Show file tree

Hide file tree

Showing 39 changed files with 114 additions and 53 deletions.
diff --git a/cfgs/aanet/aanet_sceneflow.yaml b/cfgs/aanet/aanet_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_cleanpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_cleanpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_cleanpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_cleanpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_cleanpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_cleanpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/casnet/casnet_psm_sceneflow.yaml b/cfgs/casnet/casnet_psm_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/cfnet/cfnet_sceneflow_part2.yaml b/cfgs/cfnet/cfnet_sceneflow_part2.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/coex/coex_sceneflow_amp.yaml b/cfgs/coex/coex_sceneflow_amp.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/fadnet/fadnet_sceneflow.yaml b/cfgs/fadnet/fadnet_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_cleanpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_cleanpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_cleanpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_cleanpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_cleanpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_cleanpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/gwcnet/gwcnet_sceneflow.yaml b/cfgs/gwcnet/gwcnet_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/igev/igev_kitti_amp.yaml b/cfgs/igev/igev_kitti_amp.yaml
@@ -2,7 +2,7 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: KittiDataset
             DATA_SPLIT: {
-                TRAINING: ./data/kitti12/kitti12_train194.txt,
+                TRAINING: ./data/KITTI12//kitti12_train194.txt,
                 EVALUATING: '',
                 TESTING: ''
             }
@@ -11,9 +11,9 @@ DATA_CONFIG:
 
         -   DATASET: KittiDataset
             DATA_SPLIT: {
-                TRAINING: ./data/kitti15/kitti15_train200.txt,
-                EVALUATING: ./data/kitti15/kitti15_val20.txt,
-                TESTING: ./data/kitti15/kitti15_test.txt
+                TRAINING: ./data/KITTI15//kitti15_train200.txt,
+                EVALUATING: ./data/KITTI15//kitti15_val20.txt,
+                TESTING: ./data/KITTI15//kitti15_test.txt
             }
             DATA_PATH: /yourpath/kitti15
             RETURN_RIGHT_DISP: true

diff --git a/cfgs/igev/igev_sceneflow_amp.yaml b/cfgs/igev/igev_sceneflow_amp.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/kitti15_eval.yaml b/cfgs/kitti15_eval.yaml
@@ -2,8 +2,8 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: KittiDataset
             DATA_SPLIT: {
-                EVALUATING: ./data/kitti15/kitti15_val20.txt,
-                TESTING: ./data/kitti15/kitti15_test.txt
+                EVALUATING: ./data/KITTI15//kitti15_val20.txt,
+                TESTING: ./data/KITTI15//kitti15_test.txt
             }
             DATA_PATH: /yourpath/KITTI/stereo_2015
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/msnet/msnet2d_sceneflow.yaml b/cfgs/msnet/msnet2d_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/msnet/msnet3d_sceneflow.yaml b/cfgs/msnet/msnet3d_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_finalpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_finalpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_finalpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_finalpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/psmnet/psmnet_kitti.yaml b/cfgs/psmnet/psmnet_kitti.yaml
@@ -2,7 +2,7 @@ DATA_CONFIG:
     DATA_INFOS:
 #        -   DATASET: KittiDataset
 #            DATA_SPLIT: {
-#                TRAINING: ./data/kitti12/kitti12_train194.txt,
+#                TRAINING: ./data/KITTI12//kitti12_train194.txt,
 #                EVALUATING: '',
 #                TESTING: ''
 #            }
@@ -11,9 +11,9 @@ DATA_CONFIG:
 
         -   DATASET: KittiDataset
             DATA_SPLIT: {
-                TRAINING: ./data/kitti15/kitti15_train200.txt,
-                EVALUATING: ./data/kitti15/kitti15_val20.txt,
-                TESTING: ./data/kitti15/kitti15_test.txt
+                TRAINING: ./data/KITTI15//kitti15_train200.txt,
+                EVALUATING: ./data/KITTI15//kitti15_val20.txt,
+                TESTING: ./data/KITTI15//kitti15_test.txt
             }
             DATA_PATH: /yourpath/kitti15
             RETURN_RIGHT_DISP: true

diff --git a/cfgs/psmnet/psmnet_sceneflow.yaml b/cfgs/psmnet/psmnet_sceneflow.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_cleanpass_train.txt,
-                EVALUATING: ./data/sceneflow/sceneflow_cleanpass_test.txt,
-                TESTING: ./data/sceneflow/sceneflow_cleanpass_test.txt
+                TRAINING: ./data/SceneFlow/sceneflow_cleanpass_train.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_cleanpass_test.txt,
+                TESTING: ./data/SceneFlow/sceneflow_cleanpass_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/sceneflow_eval.yaml b/cfgs/sceneflow_eval.yaml
@@ -2,7 +2,7 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset
             DATA_SPLIT: {
-                EVALUATING: ./data/sceneflow/sceneflow_finalpass_test.txt,
+                EVALUATING: ./data/SceneFlow/sceneflow_finalpass_test.txt,
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_RIGHT_DISP: false

diff --git a/cfgs/sttr/sttr_flyingthings3d.yaml b/cfgs/sttr/sttr_flyingthings3d.yaml
@@ -2,9 +2,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: FlyingThings3DSubsetDataset
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/FlyingThings3D_sttr_train.txt,
-                EVALUATING: ./data/sceneflow/FlyingThings3D_sttr_test.txt,
-                TESTING: ./data/sceneflow/FlyingThings3D_sttr_test.txt
+                TRAINING: ./data/SceneFlow/FlyingThings3D_sttr_train.txt,
+                EVALUATING: ./data/SceneFlow/FlyingThings3D_sttr_test.txt,
+                TESTING: ./data/SceneFlow/FlyingThings3D_sttr_test.txt
             }
             DATA_PATH: /yourpath/SceneFlow
             RETURN_OCC_MASK: true

diff --git a/data/kitti12/README.md → data/KITTI12/README.md b/data/kitti12/README.md → data/KITTI12/README.md
diff --git a/data/kitti12/kitti12_test.txt → data/KITTI12/kitti12_test.txt b/data/kitti12/kitti12_test.txt → data/KITTI12/kitti12_test.txt
diff --git a/data/kitti12/kitti12_train165.txt → data/KITTI12/kitti12_train165.txt b/data/kitti12/kitti12_train165.txt → data/KITTI12/kitti12_train165.txt
diff --git a/data/kitti12/kitti12_train180.txt → data/KITTI12/kitti12_train180.txt b/data/kitti12/kitti12_train180.txt → data/KITTI12/kitti12_train180.txt
diff --git a/data/kitti12/kitti12_train194.txt → data/KITTI12/kitti12_train194.txt b/data/kitti12/kitti12_train194.txt → data/KITTI12/kitti12_train194.txt
diff --git a/data/kitti12/kitti12_val14.txt → data/KITTI12/kitti12_val14.txt b/data/kitti12/kitti12_val14.txt → data/KITTI12/kitti12_val14.txt
diff --git a/data/kitti12/kitti12_val29.txt → data/KITTI12/kitti12_val29.txt b/data/kitti12/kitti12_val29.txt → data/KITTI12/kitti12_val29.txt
diff --git a/data/kitti15/README.md → data/KITTI15/README.md b/data/kitti15/README.md → data/KITTI15/README.md
diff --git a/data/kitti15/kitti15_test.txt → data/KITTI15/kitti15_test.txt b/data/kitti15/kitti15_test.txt → data/KITTI15/kitti15_test.txt
diff --git a/data/kitti15/kitti15_test_pseudo_stereo.txt → data/KITTI15/kitti15_test_pseudo_stereo.txt b/data/kitti15/kitti15_test_pseudo_stereo.txt → data/KITTI15/kitti15_test_pseudo_stereo.txt
diff --git a/data/kitti15/kitti15_train165.txt → data/KITTI15/kitti15_train165.txt b/data/kitti15/kitti15_train165.txt → data/KITTI15/kitti15_train165.txt
diff --git a/data/kitti15/kitti15_train180.txt → data/KITTI15/kitti15_train180.txt b/data/kitti15/kitti15_train180.txt → data/KITTI15/kitti15_train180.txt
diff --git a/data/kitti15/kitti15_train200.txt → data/KITTI15/kitti15_train200.txt b/data/kitti15/kitti15_train200.txt → data/KITTI15/kitti15_train200.txt
diff --git a/data/kitti15/kitti15_val20.txt → data/KITTI15/kitti15_val20.txt b/data/kitti15/kitti15_val20.txt → data/KITTI15/kitti15_val20.txt
diff --git a/data/kitti15/kitti15_val35.txt → data/KITTI15/kitti15_val35.txt b/data/kitti15/kitti15_val35.txt → data/KITTI15/kitti15_val35.txt
diff --git a/data/sceneflow/FlyingThings3D_sttr_test.txt → data/SceneFlow/FlyingThings3D_sttr_test.txt b/data/sceneflow/FlyingThings3D_sttr_test.txt → data/SceneFlow/FlyingThings3D_sttr_test.txt
diff --git a/data/sceneflow/FlyingThings3D_sttr_train.txt → data/SceneFlow/FlyingThings3D_sttr_train.txt b/data/sceneflow/FlyingThings3D_sttr_train.txt → data/SceneFlow/FlyingThings3D_sttr_train.txt
diff --git a/data/SceneFlow/README.md b/data/SceneFlow/README.md
@@ -0,0 +1,57 @@
+# Prepare SceneFlow dataset
+
+Scene Flow Datasets is available at the Scene Flow [official website](https://lmb.informatik.uni-freiburg.de/resources/datasets/SceneFlowDatasets.en.html).
+Please download the following parts of FlyingThings3D, Driving, Monkaa and place it in the `datasets/sceneflow` directory.
+- RGB images (finalpass)
+- RGB images (cleanpass)
+- Disparity
+
+And for FlyingThings3D subset, you need to download the following parts:
+- RGB images (cleanpass)
+- Disparity
+- Disparity Occlusions
+
+The directory structure should be as follows:
+```text
+data
+|   SceneFlow
+|   ├── Driving
+|   |   ├── frames_finalpass
+|   |   ├── frames_cleanpass
+|   |   ├── disparity
+...
+|   ├── Monkaa
+|   |   ├── frames_finalpass
+|   |   ├── frames_cleanpass
+|   |   ├── disparity
+...
+|   ├── FlyingThings3D
+|   |   ├── frames_finalpass
+|   |   ├── frames_cleanpass
+|   |   ├── disparity
+...
+|   ├── FlyingThings3D_subset
+|   |   ├── train
+|   |   |   ├── frames_cleanpass
+|   |   |   ├── disparity
+|   |   |   ├── disparity_occlusions
+|   |   ├── test
+|   |   |   ├── frames_cleanpass
+|   |   |   ├── disparity
+|   |   |   ├── disparity_occlusions
+```
+
+
+_Optionally you can write your own txt file and use all the parts of the dataset._ 
+
+### Reference 
+
+```bibtex
+@inproceedings{mayer2016large,
+  title={A large dataset to train convolutional networks for disparity, optical flow, and scene flow estimation},
+  author={Mayer, Nikolaus and Ilg, Eddy and Hausser, Philip and Fischer, Philipp and Cremers, Daniel and Dosovitskiy, Alexey and Brox, Thomas},
+  booktitle={Proceedings of the IEEE conference on computer vision and pattern recognition},
+  pages={4040--4048},
+  year={2016}
+}
+```
diff --git a/data/sceneflow/sceneflow_cleanpass_test.txt → data/SceneFlow/sceneflow_cleanpass_test.txt b/data/sceneflow/sceneflow_cleanpass_test.txt → data/SceneFlow/sceneflow_cleanpass_test.txt
diff --git a/data/sceneflow/sceneflow_cleanpass_train.txt → data/SceneFlow/sceneflow_cleanpass_train.txt b/data/sceneflow/sceneflow_cleanpass_train.txt → data/SceneFlow/sceneflow_cleanpass_train.txt
diff --git a/data/sceneflow/sceneflow_finalpass_test.txt → data/SceneFlow/sceneflow_finalpass_test.txt b/data/sceneflow/sceneflow_finalpass_test.txt → data/SceneFlow/sceneflow_finalpass_test.txt
diff --git a/data/sceneflow/sceneflow_finalpass_train.txt → data/SceneFlow/sceneflow_finalpass_train.txt b/data/sceneflow/sceneflow_finalpass_train.txt → data/SceneFlow/sceneflow_finalpass_train.txt
diff --git a/docs/2.prepare_dataset.md b/docs/2.prepare_dataset.md
@@ -3,9 +3,13 @@
 OpenStereo officially supports multiple datasets.
 Please follow the corresponding guidelines for data preparation.
 
-- [SceneFlow](../datasets/SceneFlow/README.md)
-- [KITTI2015](../datasets/KITTI15/README.md)
-- [KITTI2012](../datasets/KITTI12/README.md)
-- [DrivingStereo](../datasets/DrivingStereo/README.md)
-- [ETH3D](../datasets/ETH3D/README.md)
-- [Middlebury](../datasets/Middlebury/README.md)
+- [SceneFlow](../data/SceneFlow/README.md)
+- [KITTI2015](../data/KITTI15/README.md)
+- [KITTI2012](../data/KITTI12/README.md)
+- [DrivingStereo](../data/DrivingStereo/README.md)
+- [ETH3D](../data/ETH3D/README.md)
+- [Middlebury](../data/Middlebury/README.md)
+- [Sintel](../data/Sintel/README.md)
+- [FallingThings](../data/FallingThings/README.md)
+- [InStereo2K](../data/InStereo2K/README.md)
+- [UnrealStereo4k](../data/UnrealStereo4k/README.md)
diff --git a/docs/3.detailed_config.md b/docs/3.detailed_config.md
@@ -7,9 +7,9 @@ DATA_CONFIG:
     DATA_INFOS:
         -   DATASET: SceneFlowDataset  # dataset name
             DATA_SPLIT: {
-                TRAINING: ./data/sceneflow/sceneflow_cleanpass_train.txt,  # train list 
-                EVALUATING: ./data/sceneflow/sceneflow_cleanpass_test.txt,  # eval list 
-                TESTING: ./data/sceneflow/sceneflow_cleanpass_test.txt  # test list
+                TRAINING: ./data/SceneFlow/sceneflow_cleanpass_train.txt,  # train list 
+                EVALUATING: ./data/SceneFlow/sceneflow_cleanpass_test.txt,  # eval list 
+                TESTING: ./data/SceneFlow/sceneflow_cleanpass_test.txt  # test list
             }
             DATA_PATH: /yourpath/SceneFlow  # the dataset root path
             RETURN_RIGHT_DISP: false