rename data to structures (#1854)

pull/1948/head^2
谢昕辰 2022-08-03 15:43:23 +08:00 committed by MeowZheng
parent ecab73a892
commit 167f94a70b
24 changed files with 18 additions and 18 deletions

View File

@ -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

View File

@ -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()

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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)

View File

@ -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 \

View File

@ -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]

View File

@ -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()

View File

@ -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):

View File

@ -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):

View File

@ -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

View File

@ -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):

View File

@ -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):

View File

@ -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):

View File

@ -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()

View File

@ -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():

View File

@ -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