Import dataset via custom imports

.gitignore

@@ -0,0 +1,3 @@
+__pycache__**
+data/**
+pretrained_models/**

scripts/benchmark/eval2d.py

@@ -9,13 +9,11 @@
 
 from mmengine.config import Config, DictAction
 from mmengine.runner import Runner
+from mmengine.registry import RUNNERS
 
 root_dir = Path(__file__).parent.parent.parent
 sys.path.append(str(root_dir))
 
-from wildscenes.mmseg_wildscenes.registry import RUNNERS
-
-
 def parse_args():
     parser = argparse.ArgumentParser(
         description='MMSeg test (and eval) a model')

scripts/benchmark/eval3d.py

@@ -12,14 +12,13 @@
 from mmengine.config import Config, ConfigDict, DictAction
 # from mmengine.registry import RUNNERS
 from mmengine.runner import Runner
+from mmengine.registry import RUNNERS
 
 from mmdet3d.utils import replace_ceph_backend
 
 root_dir = Path(__file__).parent.parent.parent
 sys.path.append(str(root_dir))
 
-from wildscenes.mmdet_wildscenes.registry import RUNNERS3D as RUNNERS
-
 
 def parse_args():
     parser = argparse.ArgumentParser(

scripts/benchmark/train2d.py

@@ -11,12 +11,11 @@
 from mmengine.config import Config, DictAction
 from mmengine.logging import print_log
 from mmengine.runner import Runner
+from mmengine.registry import RUNNERS
 
 root_dir = Path(__file__).parent.parent.parent
 sys.path.append(str(root_dir))
 
-from wildscenes.mmseg_wildscenes.registry import RUNNERS
-
 
 def parse_args():
     parser = argparse.ArgumentParser(description='Train a segmentor on wildscenes')

scripts/benchmark/train3d.py

@@ -13,15 +13,11 @@
 from mmengine.logging import print_log
 # from mmengine.registry import RUNNERS
 from mmengine.runner import Runner
-
-from mmdet3d.utils import replace_ceph_backend
+from mmdet3d.registry import RUNNERS
 
 root_dir = Path(__file__).parent.parent.parent
 sys.path.append(str(root_dir))
 
-# from wildscenes.mmseg_wildscenes.registry.registry3d import RUNNERS
-from wildscenes.mmdet_wildscenes.registry import RUNNERS3D as RUNNERS
-
 
 def parse_args():
     parser = argparse.ArgumentParser(description='Train a 3D detector')

wildscenes/configs/_base_/datasets/wildscenes.py

@@ -1,3 +1,9 @@
+custom_imports = dict(
+    imports=[
+        'wildscenes.mmseg_wildscenes'
+    ],
+    allow_failed_imports=False)
+
 
 dataset_type = "WildscenesDataset"
 

wildscenes/configs3d/_base_/datasets/wildscenes.py

@@ -5,6 +5,11 @@
 # For labels_map we follow the uniform format of MMDetection & MMSegmentation
 # i.e. we consider the unlabeled class as the last one, which is different
 # from the original implementation of some methods e.g. Cylinder3D.
+custom_imports = dict(
+    imports=[
+        'wildscenes.mmdet_wildscenes'
+    ],
+    allow_failed_imports=False)
 dataset_type = 'WildScenesDataset3d'
 
 # Please modify the path below and replace FULLPATHTOTHISREPO with where you installed this repo