We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 425a923 commit 1dfd59dCopy full SHA for 1dfd59d
torchvision/utils.py
@@ -8,7 +8,6 @@
8
9
import numpy as np
10
import torch
11
-from packaging import version
12
from PIL import Image, ImageColor, ImageDraw, ImageFont
13
14
@@ -184,7 +183,8 @@ def _Image_fromarray(
184
183
mode paramter. See:
185
https://pillow.readthedocs.io/en/stable/releasenotes/11.3.0.html#image-fromarray-mode-parameter
186
"""
187
- if version.parse(Image.__version__) >= version.parse("11.3.0"):
+ PILLOW_VERSION = tuple(int(x) for x in PIL.__version__.split("."))
+ if PILLOW_VERSION >= (11, 3):
188
return Image.fromarray(obj)
189
else:
190
return Image.fromarray(obj, mode)
0 commit comments