From 7b765c9a2b49a4244f4e4b17fb5da2904096df70 Mon Sep 17 00:00:00 2001
From: Spencer Delcore <sdelcore@uwaterloo.ca>
Date: Fri, 17 Mar 2023 21:37:43 -0400
Subject: [PATCH] updated readme and removed hardcoded path

---
 README.md    | 4 ++++
 main_lost.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 2eef3d1..26b049c 100644
--- a/README.md
+++ b/README.md
@@ -14,6 +14,10 @@ touch __init__.py
 echo -e "import sys\nfrom os.path import dirname, join\nsys.path.insert(0, join(dirname(__file__), '.'))" >> __init__.py;
 cd -
 
+# replace $KITTI_ROOT with path to kitti
+# replace * with the script you want to run
+KITTI_ROOT=$KITTI_ROOT ./scripts/run_*.sh
+
 ```
 
 
diff --git a/main_lost.py b/main_lost.py
index c8e6b0b..967395d 100755
--- a/main_lost.py
+++ b/main_lost.py
@@ -128,7 +128,7 @@ if __name__ == "__main__":
     if args.image_path is not None:
         dataset = ImageDataset(args.image_path)
     elif args.dataset == "KITTI":
-        dataset = ImageFolderDataset("/root/kitti/training/image_2/") # TODO dont hard code
+        dataset = ImageFolderDataset("$KITTI_ROOT/training/image_2/") # TODO dont hard code
     else:
         dataset = Dataset(args.dataset, args.set, args.no_hard)
 
-- 
GitLab