Move stages to each algorithms (#1938)

* Move stages to each algorithms --------- Co-authored-by: Eunwoo Shin <eunwoo.shin@intel.com>
openvinotoolkit · Mar 24, 2023 · e654199 · e654199
1 parent a853c38
commit e654199
Show file tree

Hide file tree

Showing 100 changed files with 896 additions and 715 deletions.
diff --git a/docs/source/guide/reference/algorithm/classification/adapters/mmcls.rst b/docs/source/guide/reference/algorithm/classification/adapters/mmcls.rst
@@ -7,4 +7,56 @@ mmclassification
 
 .. automodule:: otx.algorithms.classification.adapters.mmcls
    :members:
-   :undoc-members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.exporter
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.trainer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.incremental
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.incremental.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.incremental.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.incremental.trainer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.semisl
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.semisl.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.semisl.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.classification.adapters.mmcls.task.semisl.trainer
+   :members:
+   :undoc-members:
diff --git a/docs/source/guide/reference/algorithm/detection/adapters/mmdet.rst b/docs/source/guide/reference/algorithm/detection/adapters/mmdet.rst
@@ -7,4 +7,56 @@ mmdetection
 
 .. automodule:: otx.algorithms.detection.adapters.mmdet
    :members:
-   :undoc-members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.exporter
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.trainer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.incremental
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.incremental.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.incremental.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.incremental.trainer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.semisl
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.semisl.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.semisl.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.detection.adapters.mmdet.tasks.semisl.trainer
+   :members:
+   :undoc-members:
diff --git a/docs/source/guide/reference/algorithm/segmentation/adapters/mmseg.rst b/docs/source/guide/reference/algorithm/segmentation/adapters/mmseg.rst
@@ -7,4 +7,56 @@ mmsegmentation
 
 .. automodule:: otx.algorithms.segmentation.adapters.mmseg
    :members:
-   :undoc-members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.exporter
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.trainer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.incremental
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.incremental.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.incremental.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.incremental.trainer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.semisl
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.semisl.inferrer
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.semisl.stage
+   :members:
+   :undoc-members:
+
+.. automodule:: otx.algorithms.segmentation.adapters.mmseg.tasks.semisl.trainer
+   :members:
+   :undoc-members:
diff --git a/docs/source/guide/reference/mpa/classification.rst b/docs/source/guide/reference/mpa/classification.rst
diff --git a/docs/source/guide/reference/mpa/detection.rst b/docs/source/guide/reference/mpa/detection.rst
diff --git a/docs/source/guide/reference/mpa/index.rst b/docs/source/guide/reference/mpa/index.rst
@@ -4,6 +4,5 @@ Model Preparation Algorithm
 .. toctree::
    :maxdepth: 1
 
-   classification
-   detection
-   segmentation
+   modules/index
+   utils
diff --git a/docs/source/guide/reference/mpa/segmentation.rst b/docs/source/guide/reference/mpa/segmentation.rst
diff --git a/otx/mpa/cls/__init__.py → ...fication/adapters/mmcls/tasks/__init__.py b/otx/mpa/cls/__init__.py → ...fication/adapters/mmcls/tasks/__init__.py
@@ -1,4 +1,5 @@
-# Copyright (C) 2022 Intel Corporation
+"""Initialize OTX classification tasks."""
+# Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 #
 

diff --git a/otx/mpa/cls/evaluator.py → ...ication/adapters/mmcls/tasks/evaluator.py b/otx/mpa/cls/evaluator.py → ...ication/adapters/mmcls/tasks/evaluator.py
@@ -1,11 +1,10 @@
-# Copyright (C) 2022 Intel Corporation
+"""Evaluation class for OTX Classification with MMCLS."""
+# Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 #
 
-import os.path as osp
-
+from otx.algorithms.common.adapters.mmcv.tasks.registry import STAGES
 from otx.algorithms.common.utils.logger import get_logger
-from otx.mpa.registry import STAGES
 
 from .inferrer import ClsInferrer
 
@@ -14,8 +13,10 @@
 
 @STAGES.register_module()
 class ClsEvaluator(ClsInferrer):
+    """Evaluator for MMCLS backend."""
+
     def run(self, model_cfg, model_ckpt, data_cfg, **kwargs):
-        """Run evaluation stage for classification
+        """Run evaluation stage for classification.
 
         - Run inference
         - Run evaluation via MMClassification -> MMCV