diff --git a/basicsr/utils/misc.py b/basicsr/utils/misc.py index f425d68e..1846efcc 100644 --- a/basicsr/utils/misc.py +++ b/basicsr/utils/misc.py @@ -9,7 +9,7 @@ from .dist_util import master_only from .logger import get_root_logger -IS_HIGH_VERSION = [int(m) for m in list(re.findall(r"^([0-9]+)\.([0-9]+)\.([0-9]+)([^0-9][a-zA-Z0-9]*)?(\+git.*)?$",\ +IS_HIGH_VERSION = [int(m) for m in list(re.findall(r"^(\d+)\.(\d+)\.(\d+)([\w\d\.].*)?$",\ torch.__version__)[0][:3])] >= [1, 12, 0] def gpu_is_available(): diff --git a/facelib/detection/yolov5face/face_detector.py b/facelib/detection/yolov5face/face_detector.py index 1b27e970..f489a208 100644 --- a/facelib/detection/yolov5face/face_detector.py +++ b/facelib/detection/yolov5face/face_detector.py @@ -14,11 +14,9 @@ scale_coords_landmarks, ) -# IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.')[:2])) >= (1, 9) -IS_HIGH_VERSION = [int(m) for m in list(re.findall(r"^([0-9]+)\.([0-9]+)\.([0-9]+)([^0-9][a-zA-Z0-9]*)?(\+git.*)?$",\ +IS_HIGH_VERSION = [int(m) for m in list(re.findall(r"^(\d+)\.(\d+)\.(\d+)([\w\d\.].*)?$",\ torch.__version__)[0][:3])] >= [1, 9, 0] - def isListempty(inList): if isinstance(inList, list): # Is a list return all(map(isListempty, inList))