rename data to structures (#1854)
parent
ecab73a892
commit
167f94a70b
|
@ -8,9 +8,9 @@ from mmcv.runner import load_checkpoint
|
||||||
from mmengine import Config
|
from mmengine import Config
|
||||||
from mmengine.dataset import Compose
|
from mmengine.dataset import Compose
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.models import BaseSegmentor
|
from mmseg.models import BaseSegmentor
|
||||||
from mmseg.registry import MODELS
|
from mmseg.registry import MODELS
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.utils import SampleList
|
from mmseg.utils import SampleList
|
||||||
from mmseg.visualization import SegLocalVisualizer
|
from mmseg.visualization import SegLocalVisualizer
|
||||||
|
|
||||||
|
|
|
@ -5,8 +5,8 @@ from mmcv.transforms import to_tensor
|
||||||
from mmcv.transforms.base import BaseTransform
|
from mmcv.transforms.base import BaseTransform
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.registry import TRANSFORMS
|
from mmseg.registry import TRANSFORMS
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
@TRANSFORMS.register_module()
|
@TRANSFORMS.register_module()
|
||||||
|
|
|
@ -7,8 +7,8 @@ import mmcv
|
||||||
from mmengine.hooks import Hook
|
from mmengine.hooks import Hook
|
||||||
from mmengine.runner import Runner
|
from mmengine.runner import Runner
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.registry import HOOKS
|
from mmseg.registry import HOOKS
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.visualization import SegLocalVisualizer
|
from mmseg.visualization import SegLocalVisualizer
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -7,8 +7,8 @@ import torch.nn as nn
|
||||||
from mmengine.model import BaseModule
|
from mmengine.model import BaseModule
|
||||||
from torch import Tensor
|
from torch import Tensor
|
||||||
|
|
||||||
from mmseg.data import build_pixel_sampler
|
|
||||||
from mmseg.ops import resize
|
from mmseg.ops import resize
|
||||||
|
from mmseg.structures import build_pixel_sampler
|
||||||
from mmseg.utils import ConfigType, SampleList
|
from mmseg.utils import ConfigType, SampleList
|
||||||
from ..builder import build_loss
|
from ..builder import build_loss
|
||||||
from ..losses import accuracy
|
from ..losses import accuracy
|
||||||
|
|
|
@ -4,8 +4,8 @@ import torch.nn.functional as F
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
from torch import Tensor
|
from torch import Tensor
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.registry import MODELS
|
from mmseg.registry import MODELS
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.utils import SampleList
|
from mmseg.utils import SampleList
|
||||||
from .fcn_head import FCNHead
|
from .fcn_head import FCNHead
|
||||||
|
|
||||||
|
|
|
@ -6,8 +6,8 @@ from mmengine.data import PixelData
|
||||||
from mmengine.model import BaseModel
|
from mmengine.model import BaseModel
|
||||||
from torch import Tensor
|
from torch import Tensor
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.ops import resize
|
from mmseg.ops import resize
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.utils import (ForwardResults, OptConfigType, OptMultiConfig,
|
from mmseg.utils import (ForwardResults, OptConfigType, OptMultiConfig,
|
||||||
OptSampleList, SampleList)
|
OptSampleList, SampleList)
|
||||||
|
|
||||||
|
|
|
@ -16,11 +16,11 @@ def register_all_modules(init_default_scope: bool = True) -> None:
|
||||||
to https://github.com/open-mmlab/mmengine/blob/main/docs/en/tutorials/registry.md
|
to https://github.com/open-mmlab/mmengine/blob/main/docs/en/tutorials/registry.md
|
||||||
Defaults to True.
|
Defaults to True.
|
||||||
""" # noqa
|
""" # noqa
|
||||||
import mmseg.data # noqa: F401,F403
|
|
||||||
import mmseg.datasets # noqa: F401,F403
|
import mmseg.datasets # noqa: F401,F403
|
||||||
import mmseg.engine # noqa: F401,F403
|
import mmseg.engine # noqa: F401,F403
|
||||||
import mmseg.evaluation # noqa: F401,F403
|
import mmseg.evaluation # noqa: F401,F403
|
||||||
import mmseg.models # noqa: F401,F403
|
import mmseg.models # noqa: F401,F403
|
||||||
|
import mmseg.structures # noqa: F401,F403
|
||||||
|
|
||||||
if init_default_scope:
|
if init_default_scope:
|
||||||
never_created = DefaultScope.get_current_instance() is None \
|
never_created = DefaultScope.get_current_instance() is None \
|
||||||
|
|
|
@ -5,7 +5,7 @@ from typing import Dict, List, Optional, Sequence, Tuple, Union
|
||||||
import torch
|
import torch
|
||||||
from mmengine.config import ConfigDict
|
from mmengine.config import ConfigDict
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
# Type hint of config data
|
# Type hint of config data
|
||||||
ConfigType = Union[ConfigDict, dict]
|
ConfigType = Union[ConfigDict, dict]
|
||||||
|
|
|
@ -6,8 +6,8 @@ from mmengine import Visualizer
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
from mmengine.dist import master_only
|
from mmengine.dist import master_only
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.registry import VISUALIZERS
|
from mmseg.registry import VISUALIZERS
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
@VISUALIZERS.register_module()
|
@VISUALIZERS.register_module()
|
||||||
|
|
|
@ -6,7 +6,7 @@ import pytest
|
||||||
import torch
|
import torch
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
def _equal(a, b):
|
def _equal(a, b):
|
||||||
|
|
|
@ -6,8 +6,8 @@ import unittest
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from mmengine.data import BaseDataElement
|
from mmengine.data import BaseDataElement
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.datasets.transforms import PackSegInputs
|
from mmseg.datasets.transforms import PackSegInputs
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
class TestPackSegInputs(unittest.TestCase):
|
class TestPackSegInputs(unittest.TestCase):
|
||||||
|
|
|
@ -5,8 +5,8 @@ from unittest.mock import Mock
|
||||||
import torch
|
import torch
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.engine.hooks import SegVisualizationHook
|
from mmseg.engine.hooks import SegVisualizationHook
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.visualization import SegLocalVisualizer
|
from mmseg.visualization import SegLocalVisualizer
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -5,8 +5,8 @@ import numpy as np
|
||||||
import torch
|
import torch
|
||||||
from mmengine.data import BaseDataElement, PixelData
|
from mmengine.data import BaseDataElement, PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.evaluation import CitysMetric
|
from mmseg.evaluation import CitysMetric
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
class TestCitysMetric(TestCase):
|
class TestCitysMetric(TestCase):
|
||||||
|
|
|
@ -5,8 +5,8 @@ import numpy as np
|
||||||
import torch
|
import torch
|
||||||
from mmengine.data import BaseDataElement, PixelData
|
from mmengine.data import BaseDataElement, PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.evaluation import IoUMetric
|
from mmseg.evaluation import IoUMetric
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
class TestIoUMetric(TestCase):
|
class TestIoUMetric(TestCase):
|
||||||
|
|
|
@ -4,8 +4,8 @@ from unittest import TestCase
|
||||||
import torch
|
import torch
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
|
||||||
from mmseg.models import SegDataPreProcessor
|
from mmseg.models import SegDataPreProcessor
|
||||||
|
from mmseg.structures import SegDataSample
|
||||||
|
|
||||||
|
|
||||||
class TestSegDataPreProcessor(TestCase):
|
class TestSegDataPreProcessor(TestCase):
|
||||||
|
|
|
@ -13,7 +13,7 @@ from mmcv.cnn.utils import revert_sync_batchnorm
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
from torch import Tensor
|
from torch import Tensor
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.utils import register_all_modules
|
from mmseg.utils import register_all_modules
|
||||||
|
|
||||||
register_all_modules()
|
register_all_modules()
|
||||||
|
|
|
@ -2,8 +2,8 @@
|
||||||
import pytest
|
import pytest
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
from mmseg.data import OHEMPixelSampler
|
|
||||||
from mmseg.models.decode_heads import FCNHead
|
from mmseg.models.decode_heads import FCNHead
|
||||||
|
from mmseg.structures import OHEMPixelSampler
|
||||||
|
|
||||||
|
|
||||||
def _context_for_ohem():
|
def _context_for_ohem():
|
||||||
|
|
|
@ -10,7 +10,7 @@ import numpy as np
|
||||||
import torch
|
import torch
|
||||||
from mmengine.data import PixelData
|
from mmengine.data import PixelData
|
||||||
|
|
||||||
from mmseg.data import SegDataSample
|
from mmseg.structures import SegDataSample
|
||||||
from mmseg.visualization import SegLocalVisualizer
|
from mmseg.visualization import SegLocalVisualizer
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue