unlock albumentations version (#14746)

This commit is contained in:
Wang Xin 2025-02-25 16:30:08 +08:00 committed by GitHub
parent 2b7b76310b
commit 37f44372b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 14 additions and 13 deletions

View File

@ -24,14 +24,16 @@ import numpy as np
import albumentations as A import albumentations as A
from albumentations.core.transforms_interface import DualTransform from albumentations.core.transforms_interface import DualTransform
from albumentations.augmentations.geometric import functional as fgeometric from albumentations.augmentations.geometric import functional as fgeometric
from packaging import version
ALBU_VERSION = version.parse(A.__version__)
IS_ALBU_NEW_VERSION = ALBU_VERSION >= version.parse("1.4.15")
# Custom resize transformation mimicking Imgaug's behavior with scaling # Custom resize transformation mimicking Imgaug's behavior with scaling
class ImgaugLikeResize(DualTransform): class ImgaugLikeResize(DualTransform):
def __init__( def __init__(self, scale_range=(0.5, 3.0), interpolation=1, p=1.0):
self, scale_range=(0.5, 3.0), interpolation=1, always_apply=False, p=1.0 super(ImgaugLikeResize, self).__init__(p)
):
super(ImgaugLikeResize, self).__init__(always_apply, p)
self.scale_range = scale_range self.scale_range = scale_range
self.interpolation = interpolation self.interpolation = interpolation
@ -41,11 +43,10 @@ class ImgaugLikeResize(DualTransform):
new_height = int(height * scale) new_height = int(height * scale)
new_width = int(width * scale) new_width = int(width * scale)
# For compatibility with Albumentations 1.4.15 and later if IS_ALBU_NEW_VERSION:
# return fgeometric.resize( return fgeometric.resize(
# img, (new_height, new_width), interpolation=self.interpolation img, (new_height, new_width), interpolation=self.interpolation
# ) )
return fgeometric.resize( return fgeometric.resize(
img, new_height, new_width, interpolation=self.interpolation img, new_height, new_width, interpolation=self.interpolation
) )

View File

@ -56,9 +56,9 @@ dependencies = [
"fonttools>=4.24.0", "fonttools>=4.24.0",
"fire>=0.3.0", "fire>=0.3.0",
"requests", "requests",
"albumentations==1.4.10", "albumentations",
# to be compatible with albumentations # to be compatible with albumentations
"albucore==0.0.13" "albucore"
] ]
[project.urls] [project.urls]

View File

@ -11,7 +11,7 @@ cython
Pillow Pillow
pyyaml pyyaml
requests requests
albumentations==1.4.10 albumentations
# to be compatible with albumentations # to be compatible with albumentations
albucore==0.0.13 albucore
packaging packaging