Pillowの最新版7.0.0だとpytorchのtransformが死にます.

タイトルのままでです.

pillowの最新版 7.0.0 にアップグレードしてpytorchのtransformが含まれているコードを動かすと以下のようになります.

  File "train.py", line 6, in <module>
    from utils.datasets import *
  File "/home/jun/Desktop/work_space/sportip/hogehoge/utils/datasets.py", line 12, in <module>
    import torchvision.transforms as transforms
  File "/home/jun/.local/share/virtualenvs/hogehoge-IBQiZg8J/lib/python3.6/site-packages/torchvision/__init__.py", line 4, in <module>
    from torchvision import datasets
  File "/home/jun/.local/share/virtualenvs/hogehoge-IBQiZg8J/lib/python3.6/site-packages/torchvision/datasets/__init__.py", line 9, in <module>
    from .fakedata import FakeData
  File "/home/jun/.local/share/virtualenvs/hogehoge-IBQiZg8J/lib/python3.6/site-packages/torchvision/datasets/fakedata.py", line 3, in <module>
    from .. import transforms
  File "/home/jun/.local/share/virtualenvs/hogehoge-IBQiZg8J/lib/python3.6/site-packages/torchvision/transforms/__init__.py", line 1, in <module>
    from .transforms import *
  File "/home/jun/.local/share/virtualenvs/hogehoge-IBQiZg8J/lib/python3.6/site-packages/torchvision/transforms/transforms.py", line 17, in <module>
    from . import functional as F
  File "/home/jun/.local/share/virtualenvs/hogehoge-IBQiZg8J/lib/python3.6/site-packages/torchvision/transforms/functional.py", line 5, in <module>
    from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION
ImportError: cannot import name 'PILLOW_VERSION'

これの原因は 2020年にリリースされた pillow バージョン7.0.0 のせいです.

これを回避するためにはpillowのインストール時に前のバージョンを指定すればいいです.

pipenv install pillow==6.2.1