typing_utils.py 805 Bytes
Newer Older
1
2
# Copyright (c) OpenMMLab. All rights reserved.
"""Collecting some commonly used type hint in MMDetection3D."""
zhangshilong's avatar
zhangshilong committed
3
from typing import List, Optional, Union
4

5
from mmdet.models.task_modules.samplers import SamplingResult
6
from mmengine.config import ConfigDict
7
from mmengine.structures import InstanceData
8

9
from mmdet3d.structures.det3d_data_sample import Det3DDataSample
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

# Type hint of config data
ConfigType = Union[ConfigDict, dict]
OptConfigType = Optional[ConfigType]

# Type hint of one or more config data
MultiConfig = Union[ConfigType, List[ConfigType]]
OptMultiConfig = Optional[MultiConfig]

InstanceList = List[InstanceData]
OptInstanceList = Optional[InstanceList]

SamplingResultList = List[SamplingResult]

OptSamplingResultList = Optional[SamplingResultList]
25
SampleList = List[Det3DDataSample]