diff --git a/projects/CenterNet2/centernet/__init__.py b/projects/CenterNet2/centernet/__init__.py index e17db317..e07111d4 100644 --- a/projects/CenterNet2/centernet/__init__.py +++ b/projects/CenterNet2/centernet/__init__.py @@ -12,3 +12,6 @@ from .data.datasets.objects365 import categories_v1 from .data.datasets.coco import _PREDEFINED_SPLITS_COCO from .data.datasets import nuimages + +# Importing function from config.py +from .config import add_centernet_config