fix PIL sample mode deprecated warning (#42307)

* fix PIL sample mode deprecated warning * compatible with old pil version

fix PIL sample mode deprecated warning (#42307)
* fix PIL sample mode deprecated warning * compatible with old pil version
c7a258fe · LielinJiang · GitHub · 9fd2c546 · c7a258fe
隐藏空白更改
内联并排

Showing with 19 addition and 8 deletion

python/paddle/vision/transforms/functional_pil.py python/paddle/vision/transforms/functional_pil.py +19 -8

未找到文件。
--- a/python/paddle/vision/transforms/functional_pil.py
+++ b/python/paddle/vision/transforms/functional_pil.py
@@ -32,14 +32,25 @@ else:
    Sequence = collections.abc.Sequence
    Iterable = collections.abc.Iterable
-_pil_interp_from_str = {
+try:
-    'nearest': Image.NEAREST,
+    # PIL version >= "9.1.0"
-    'bilinear': Image.BILINEAR,
+    _pil_interp_from_str = {
-    'bicubic': Image.BICUBIC,
+        'nearest': Image.Resampling.NEAREST,
-    'box': Image.BOX,
+        'bilinear': Image.Resampling.BILINEAR,
-    'lanczos': Image.LANCZOS,
+        'bicubic': Image.Resampling.BICUBIC,
-    'hamming': Image.HAMMING
+        'box': Image.Resampling.BOX,
-}
+        'lanczos': Image.Resampling.LANCZOS,
+        'hamming': Image.Resampling.HAMMING
+    }
+except:
+    _pil_interp_from_str = {
+        'nearest': Image.NEAREST,
+        'bilinear': Image.BILINEAR,
+        'bicubic': Image.BICUBIC,
+        'box': Image.BOX,
+        'lanczos': Image.LANCZOS,
+        'hamming': Image.HAMMING
+    }
 __all__ = []