2020-08-11 19:23:35 +08:00
|
|
|
import mmcv
|
2020-07-07 20:52:19 +08:00
|
|
|
|
2020-08-11 19:23:35 +08:00
|
|
|
from .version import __version__, version_info
|
|
|
|
|
2021-07-02 00:41:52 +08:00
|
|
|
MMCV_MIN = '1.3.7'
|
2021-03-31 08:55:09 +08:00
|
|
|
MMCV_MAX = '1.4.0'
|
2020-08-11 19:23:35 +08:00
|
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
|
|
|
|
|
|
|
mmcv_min_version = digit_version(MMCV_MIN)
|
|
|
|
mmcv_max_version = digit_version(MMCV_MAX)
|
|
|
|
mmcv_version = digit_version(mmcv.__version__)
|
|
|
|
|
|
|
|
|
|
|
|
assert (mmcv_min_version <= mmcv_version <= mmcv_max_version), \
|
|
|
|
f'MMCV=={mmcv.__version__} is used but incompatible. ' \
|
|
|
|
f'Please install mmcv>={mmcv_min_version}, <={mmcv_max_version}.'
|
|
|
|
|
|
|
|
__all__ = ['__version__', 'version_info']
|