From 8c4920e2014d4818ec3ecafb7b526d896a659d1c Mon Sep 17 00:00:00 2001 From: shenyunhang Date: Wed, 8 May 2024 09:40:44 +0800 Subject: [PATCH] fix dataset root --- ape/data/datasets/odinw_instance.py | 2 +- ape/data/datasets/register_bdd100k_semseg.py | 2 +- ape/data/datasets/seginw_instance.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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)