Segmentation refactoring

CHANGELOG.md

@@ -14,6 +14,7 @@ All notable changes to this project will be documented in this file.
 - Enhance DetCon logic and SupCon for semantic segmentation(<https://github.com/openvinotoolkit/training_extensions/pull/1958>)
 - Classification task refactoring (<https://github.com/openvinotoolkit/training_extensions/pull/1972>)
 - Extend OTX explain CLI (<https://github.com/openvinotoolkit/training_extensions/pull/1941>)
+- Segmentation task refactoring (<https://github.com/openvinotoolkit/training_extensions/pull/1977>)
 
 ### Bug fixes
 

docs/source/guide/reference/algorithm/segmentation/tasks.rst

@@ -5,6 +5,6 @@ Tasks
    :maxdepth: 3
    :caption: Contents:
 
-.. automodule:: otx.algorithms.segmentation.tasks
+.. automodule:: otx.algorithms.segmentation.task
    :members:
    :undoc-members:

otx/algorithms/common/tasks/base_task.py

@@ -96,6 +96,7 @@ class OTXTask(IInferenceTask, IExportTask, IEvaluationTask, IUnload, ABC):
     def __init__(self, task_environment: TaskEnvironment, output_path: Optional[str] = None):
         self._config: Dict[Any, Any] = {}
         self._task_environment = task_environment
+        self._model_name = task_environment.model_template.name
         self._task_type = task_environment.model_template.task_type
         self._labels = task_environment.get_labels(include_empty=False)
         self._work_dir_is_temp = False

otx/algorithms/segmentation/__init__.py

@@ -2,3 +2,7 @@
 
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
+
+from .task import OTXSegmentationTask
+
+__all__ = ["OTXSegmentationTask"]

otx/algorithms/segmentation/adapters/mmseg/__init__.py

@@ -15,7 +15,6 @@
 # See the License for the specific language governing permissions
 # and limitations under the License.
 
-
 from .datasets import MPASegDataset
 from .models import (
     ClassIncrEncoderDecoder,
@@ -33,6 +32,7 @@
     StepScalarScheduler,
     SupConDetConB,
 )
+from .nncf import CustomstepLrUpdaterHook
 
 # fmt: off
 # isort: off
@@ -49,6 +49,7 @@
     "LiteHRNet",
     "MMOVBackbone",
     "CustomFCNHead",
+    "CustomstepLrUpdaterHook",
     "MMOVDecodeHead",
     "DetConLoss",
     "SelfSLMLP",