OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
909 B
29 lines
909 B
# Copyright (c) OpenMMLab. All rights reserved. |
|
import mmcv |
|
|
|
from .version import __version__, short_version |
|
|
|
|
|
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_minimum_version = '1.3.17' |
|
mmcv_maximum_version = '1.5.0' |
|
mmcv_version = digit_version(mmcv.__version__) |
|
|
|
|
|
assert (mmcv_version >= digit_version(mmcv_minimum_version) |
|
and mmcv_version <= digit_version(mmcv_maximum_version)), \ |
|
f'MMCV=={mmcv.__version__} is used but incompatible. ' \ |
|
f'Please install mmcv>={mmcv_minimum_version}, <={mmcv_maximum_version}.' |
|
|
|
__all__ = ['__version__', 'short_version']
|
|
|