mirror of https://github.com/alibaba/EasyCV.git
31 lines
704 B
Python
31 lines
704 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
import sys
|
|
from distutils.version import LooseVersion
|
|
|
|
import numpy as np
|
|
|
|
|
|
def check_numpy():
|
|
# if use xtcocotools which support lower version of numpy
|
|
# skip check
|
|
try:
|
|
import xtcocotools
|
|
except ModuleNotFoundError:
|
|
return
|
|
|
|
try:
|
|
from xtcocotools.coco import COCO
|
|
return
|
|
except ValueError as e:
|
|
pass
|
|
|
|
def require(version):
|
|
if LooseVersion(np.__version__) < LooseVersion(version):
|
|
raise ImportError(
|
|
f'numpy version should be greater than {version}')
|
|
|
|
if sys.version_info.minor == 6:
|
|
require('1.19.5')
|
|
else:
|
|
require('1.20.0')
|