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/adapters/mmcv/hooks/__init__.py

@@ -35,6 +35,7 @@
 from .fp16_sam_optimizer_hook import Fp16SAMOptimizerHook
 from .ib_loss_hook import IBLossHook
 from .logger_hook import LoggerReplaceHook, OTXLoggerHook
+from .lr_updater_hook import CustomstepLrUpdaterHook
 from .model_ema_v2_hook import ModelEmaV2Hook
 from .no_bias_decay_hook import NoBiasDecayHook
 from .progress_hook import OTXProgressHook
@@ -55,6 +56,7 @@
     "AdaptiveTrainSchedulingHook",
     "CancelInterfaceHook",
     "CancelTrainingHook",
+    "CustomstepLrUpdaterHook",
     "CheckpointHookWithValResults",
     "EnsureCorrectBestCheckpointHook",
     "ComposedDataLoadersHook",

otx/algorithms/segmentation/adapters/mmseg/nncf/hooks.py → otx/algorithms/common/adapters/mmcv/hooks/lr_updater_hook.py

@@ -1,4 +1,4 @@
-"""NNCF task related hooks."""
+"""Module for defining LrUpdaterHook and CustomLRUpdateHook for self-supervised learning using mmseg."""
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 #

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,