|
a |
|
b/ViTPose/mmpose/__init__.py |
|
|
1 |
# Copyright (c) OpenMMLab. All rights reserved. |
|
|
2 |
import mmcv |
|
|
3 |
|
|
|
4 |
from .version import __version__, short_version |
|
|
5 |
|
|
|
6 |
|
|
|
7 |
def digit_version(version_str): |
|
|
8 |
digit_version = [] |
|
|
9 |
for x in version_str.split('.'): |
|
|
10 |
if x.isdigit(): |
|
|
11 |
digit_version.append(int(x)) |
|
|
12 |
elif x.find('rc') != -1: |
|
|
13 |
patch_version = x.split('rc') |
|
|
14 |
digit_version.append(int(patch_version[0]) - 1) |
|
|
15 |
digit_version.append(int(patch_version[1])) |
|
|
16 |
return digit_version |
|
|
17 |
|
|
|
18 |
|
|
|
19 |
mmcv_minimum_version = '1.3.8' |
|
|
20 |
mmcv_maximum_version = '1.5.0' |
|
|
21 |
mmcv_version = digit_version(mmcv.__version__) |
|
|
22 |
|
|
|
23 |
|
|
|
24 |
assert (mmcv_version >= digit_version(mmcv_minimum_version) |
|
|
25 |
and mmcv_version <= digit_version(mmcv_maximum_version)), \ |
|
|
26 |
f'MMCV=={mmcv.__version__} is used but incompatible. ' \ |
|
|
27 |
f'Please install mmcv>={mmcv_minimum_version}, <={mmcv_maximum_version}.' |
|
|
28 |
|
|
|
29 |
__all__ = ['__version__', 'short_version'] |