diff --git a/ape/data/datasets/odinw_instance.py b/ape/data/datasets/odinw_instance.py index 1e0ebea..9a15374 100644 --- 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 index 2eee0ed..25fc716 100644 --- 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 index 9c67f30..f9c487f 100644 --- 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)