__init__.py 1.65 KB
Newer Older
1
2
3
import mmcv

import mmdet
4
import mmseg
zhangwenwei's avatar
zhangwenwei committed
5
6
from .version import __version__, short_version

7
8
9
10
11
12
13
14
15
16
17
18
19

def digit_version(version_str):
    digit_version = []
    for x in version_str.split('.'):
        if x.isdigit():
            digit_version.append(int(x))
        elif x.find('rc') != -1:
            patch_version = x.split('rc')
            digit_version.append(int(patch_version[0]) - 1)
            digit_version.append(int(patch_version[1]))
    return digit_version


20
mmcv_minimum_version = '1.3.8'
xiliu8006's avatar
xiliu8006 committed
21
mmcv_maximum_version = '1.4.0'
22
23
24
25
mmcv_version = digit_version(mmcv.__version__)


assert (mmcv_version >= digit_version(mmcv_minimum_version)
twang's avatar
twang committed
26
        and mmcv_version <= digit_version(mmcv_maximum_version)), \
27
    f'MMCV=={mmcv.__version__} is used but incompatible. ' \
twang's avatar
twang committed
28
    f'Please install mmcv>={mmcv_minimum_version}, <={mmcv_maximum_version}.'
29

30
31
mmdet_minimum_version = '2.14.0'
mmdet_maximum_version = '3.0.0'
32
33
34
35
36
37
38
mmdet_version = digit_version(mmdet.__version__)
assert (mmdet_version >= digit_version(mmdet_minimum_version)
        and mmdet_version <= digit_version(mmdet_maximum_version)), \
    f'MMDET=={mmdet.__version__} is used but incompatible. ' \
    f'Please install mmdet>={mmdet_minimum_version}, ' \
    f'<={mmdet_maximum_version}.'

39
40
mmseg_minimum_version = '0.14.1'
mmseg_maximum_version = '1.0.0'
41
42
43
44
45
46
47
mmseg_version = digit_version(mmseg.__version__)
assert (mmseg_version >= digit_version(mmseg_minimum_version)
        and mmseg_version <= digit_version(mmseg_maximum_version)), \
    f'MMSEG=={mmseg.__version__} is used but incompatible. ' \
    f'Please install mmseg>={mmseg_minimum_version}, ' \
    f'<={mmseg_maximum_version}.'

zhangwenwei's avatar
zhangwenwei committed
48
__all__ = ['__version__', 'short_version']