fix(opendataset): correct wrong catalog paths for two VOC2012 datasets

correct the catalog paths for VOC2012Detection and VOC2012ActionClassification PR Closed: #857
Graviti-AI · Jul 27, 2021 · 7d7b3bd · 7d7b3bd
1 parent 0cb35ba
commit 7d7b3bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorbay/opendataset/VOC2012ActionClassification/loader.py b/tensorbay/opendataset/VOC2012ActionClassification/loader.py
@@ -53,7 +53,7 @@ def VOC2012ActionClassification(path: str) -> Dataset:
     action_path = os.path.join(path, "ImageSets", "Action")
 
     dataset = Dataset(DATASET_NAME)
-    dataset.load_catalog("catalog.json")
+    dataset.load_catalog(os.path.join(os.path.dirname(__file__), "catalog.json"))
 
     for segment_name in _SEGMENT_NAMES:
         segment = dataset.create_segment(segment_name)

diff --git a/tensorbay/opendataset/VOC2012Detection/loader.py b/tensorbay/opendataset/VOC2012Detection/loader.py
@@ -54,7 +54,7 @@ def VOC2012Detection(path: str) -> Dataset:
     main_path = os.path.join(path, "ImageSets", "Main")
 
     dataset = Dataset(DATASET_NAME)
-    dataset.load_catalog("catalog.json")
+    dataset.load_catalog(os.path.join(os.path.dirname(__file__), "catalog.json"))
 
     for segment_name in _SEGMENT_NAMES:
         segment = dataset.create_segment(segment_name)