Expose learner and dataset classes (issue #132)

opendr-eu · Oct 5, 2021 · f456330 · f456330
1 parent 1433d40
commit f456330
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/opendr/perception/panoptic_segmentation/__init__.py b/src/opendr/perception/panoptic_segmentation/__init__.py
@@ -0,0 +1,4 @@
+from opendr.perception.panoptic_segmentation.datasets import CityscapesDataset, KittiDataset
+from opendr.perception.panoptic_segmentation.efficient_ps import EfficientPsLearner
+
+__all__ = ['CityscapesDataset', 'KittiDataset', 'EfficientPsLearner']
diff --git a/src/opendr/perception/panoptic_segmentation/efficient_ps/example_usage.py b/src/opendr/perception/panoptic_segmentation/efficient_ps/example_usage.py
@@ -16,8 +16,7 @@
 import cv2
 
 from opendr.engine.data import Image
-from opendr.perception.panoptic_segmentation.datasets import CityscapesDataset, KittiDataset
-from opendr.perception.panoptic_segmentation.efficient_ps import EfficientPsLearner
+from opendr.perception.panoptic_segmentation import EfficientPsLearner, CityscapesDataset, KittiDataset
 
 DATA_ROOT = '/home/USER/data/efficientPS'
 CITYSCAPES_ROOT = f'{DATA_ROOT}/converted_datasets/cityscapes'

diff --git a/tests/sources/tools/perception/panoptic_segmentation/efficient_ps/test_efficient_ps.py b/tests/sources/tools/perception/panoptic_segmentation/efficient_ps/test_efficient_ps.py
@@ -21,8 +21,7 @@
 
 from opendr.engine.data import Image
 from opendr.engine.target import Heatmap
-from opendr.perception.panoptic_segmentation.datasets import CityscapesDataset
-from opendr.perception.panoptic_segmentation.efficient_ps import EfficientPsLearner
+from opendr.perception.panoptic_segmentation import EfficientPsLearner, CityscapesDataset
 
 
 def rmfile(path):