fix dataset root

shenyunhang · May 8, 2024 · 8c4920e · 8c4920e
1 parent 3bc4e5c
commit 8c4920e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ape/data/datasets/odinw_instance.py b/ape/data/datasets/odinw_instance.py
@@ -820,5 +820,5 @@ def register_all_odinw(root):
 
 if __name__.endswith(".odinw_instance"):
  # Assume pre-defined datasets live in `./datasets`.
- _root = os.getenv("DATASET", "datasets")
+ _root = os.getenv("DETECTRON2_DATASETS", "datasets")
  register_all_odinw(_root)
diff --git a/ape/data/datasets/register_bdd100k_semseg.py b/ape/data/datasets/register_bdd100k_semseg.py
@@ -94,5 +94,5 @@ def register_all_bdd_semseg(root):
 
 if __name__.endswith(".register_bdd100k_semseg"):
  # Assume pre-defined datasets live in `./datasets`.
- _root = os.getenv("DATASET", "datasets")
+ _root = os.getenv("DETECTRON2_DATASETS", "datasets")
  register_all_bdd_semseg(_root)
diff --git a/ape/data/datasets/seginw_instance.py b/ape/data/datasets/seginw_instance.py
@@ -138,5 +138,5 @@ def register_all_seginw(root):
 
 if __name__.endswith(".seginw_instance"):
  # Assume pre-defined datasets live in `./datasets`.
- _root = os.getenv("DATASET", "datasets")
+ _root = os.getenv("DETECTRON2_DATASETS", "datasets")
  register_all_seginw(_root)