ImportError: cannot import name 'PILLOW_VERSION'

torchvision과 Pillow를 같이 쓸 때 이런 에러가 나는 경우가 있다.

인터넷을 뒤져보면 pillow의 버전을 6.1로 낮추면 해결된다는 얘기가 있는데 내 경우는 해결이 안됐다.

 

내 경우엔 torchvision/transforms/functional.py 파일에서 

'from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION' 을 하고 있었다.

 

한편 PIL/__init__.py 파일에는 

 

# VERSION was removed in Pillow 6.0.0.
# PILLOW_VERSION was removed in Pillow 7.0.0.
# Use __version__ instead.
__version__ = _version.__version__

 

로 적혀있었다.

 

그래서 functional.py 파일을 다음과 같이 바꾸었더니 해결되었다.

 

#from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION

from PIL import Image, ImageOps, ImageEnhance, __version__

 

물론 Pillow는 7.0.0을 사용해도 된다 ^^