Fix KITTI loader and citation

divyansh7877 · Jul 1, 2022 · 4f224a8 · 4f224a8
1 parent aa0dbee
commit 4f224a8
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -107,11 +107,13 @@ python stereo.py
 
 If you find our work useful in your research, please consider citing our paper
 
-    @inproceedings{bangunharcana2021coex,
+    @inproceedings{bangunharcana2021correlate,
       title={Correlate-and-Excite: Real-Time Stereo Matching via Guided Cost Volume Excitation},
-      author={Bangunharcana, Antyanta and Cho, Jae Won and Lee, Seokju and Kweon, In So and Kim, Kyung-Soo and Soohyun Kim},
-      booktitle={IROS},
-      year={2021}
+      author={Bangunharcana, Antyanta and Cho, Jae Won and Lee, Seokju and Kweon, In So and Kim, Kyung-Soo and Kim, Soohyun},
+      booktitle={2021 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)},
+      pages={3542--3548},
+      year={2021},
+      organization={IEEE}
     }
 
 ## Acknowledgements

diff --git a/dataloaders/stereo/KITTIloader2012.py b/dataloaders/stereo/KITTIloader2012.py
@@ -38,7 +38,7 @@ def dataloader(filepath, returnCalib=False, trainAll=False):
     disp_val_L = []
     calib_val = []
     for i in range(len(test_list)):
-        name = train_list[i].split('.')[0] + '.png'
+        name = test_list[i].split('.')[0] + '.png'
         left_val.append(filepath + '/colored_0/' + name)
         right_val.append(filepath + '/colored_1/' + name)
         disp_val_L.append(filepath + '/disp_noc/' + name)

diff --git a/dataloaders/stereo/KITTIloader2015.py b/dataloaders/stereo/KITTIloader2015.py
@@ -38,7 +38,7 @@ def dataloader(filepath, returnCalib=False, trainAll=False):
     disp_val_L = []
     calib_val = []
     for i in range(len(test_list)):
-        name = train_list[i].split('.')[0] + '.png'
+        name = test_list[i].split('.')[0] + '.png'
         left_val.append(filepath + '/image_2/' + name)
         right_val.append(filepath + '/image_3/' + name)
         disp_val_L.append(filepath + '/disp_occ_0/' + name)