기본 콘텐츠로 건너뛰기

ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'


Deep Learning을 공부하면 pytoch 사용은 필수죠. 또한 데이터 사용을 위해서는 torchvision을 필수 package입니다. 


Deep Learning에서 Hello World인 MNIST를 학습하기 위해 torchvision package에서 datasets, transforms를 import하면 다음과 같은 에러가 발생하여 해결책을 공유합니다.


Code 및 Error Message

from torchvision import datasets, transforms

from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION
from PIL import PILLOW_VERSION

ImportError: cannot import name 'PILLOW_VERSION' from 'PIL' (C:\Users\admin\Anaconda3\envs\pytorch\lib\site-packages\PIL\__init__.py)

PILLOW_VERSION 를 직접 import해도 같은 에러가 발생했습니다.

from PIL import PILLOW_VERSION

from PIL import PILLOW_VERSION

ImportError: cannot import name 'PILLOW_VERSION' from 'PIL' (C:\Users\ynebu\Anaconda3\envs\pytorch\lib\site-packages\PIL\__init__.py)

Error Caused

Pillow 버젼 문제로 생각되어 업그래이드를 진행했으나  전 이미 최신 버젼이 설치된 상태였습니다.

pip install --upgrade Pillow

Requirement already up-to-date: Pillow in c:\users\ynebu\anaconda3\envs\pytorch\lib\site-packages (7.0.0)

그래서 __init__.py 파일을 확인해 보니 PILLOW_VERSION은 7.0.0 버젼부터 삭제된걸 확인했습니다. 대신 __version__를 사용하라고 가이드 되어있습니다.

C:\Users\admin\Anaconda3\envs\pytorch\Lib\site-packages\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__ 

Error Solution

https://github.com/pytorch/vision/issues/1712
위 git은 지금 저희가 겪는 문제에 대해서 논하는 글 입니다.
"New Pillow version (7.0.0) breaks torchvision (ImportError: cannot import name 'PILLOW_VERSION' from 'PIL')"
9일전(2020년 1월2일) 포스팅 되었고, 답글을 보면 차주에 새 버젼이 나온다고 했는데 아직 반영 전인거 같네요

해결 방법은 두 개가 있다고 생각됩니다.
첫 번째 방법은 제가 확인은 하지않았지만 위 git에서 많은 사람들이 가이드한 내용입니다. Pillow version을 6점대로 내리면 됩니다. 그럼 PILLOW_VERSION 변수명을 참조할 수 있어 에러가 발생 안합니다.

conda install -c conda-forge pillow"<7.0.0"

두 번째 해결책은 제가 적용하고 git comment로 추가한 내용입니다. torchvision에서 Pillow를 참조하는 부분을 수정하는 방법입니다.
대상파일인 functional.py을 열고, PILLOW_VERSION을 지우고 __version__으로 변경하시면 됩니다.
C:\Users\admin\Anaconda3\envs\pytorch\Lib\site-packages\torchvision\transforms\functional.py

#from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION

from PIL import Image, ImageOps, ImageEnhance, version 

감사합니다.


댓글

이 블로그의 인기 게시물

[Deep Learning-딥러닝] 신경망 구조

뉴런 표현 및 연산 방법 생물학의 신경 세포를 단순화하여 모델링 한것이 뉴런입니다.  뉴런은 신경망의 기본 단위 입니다. 뉴런은 여러 신호를 받아, 하나의 신호를 만들어 전달하는 역할을 합니다. 출력을 내기 전에 활성 함수(activation function)을 통해서 비선형 특성을 가할 수 있습니다. 뉴런 연산 방법은 다음과 같습니다. 두 벡터  가중치 weight와 입력 x의  내적 을 구한 후 모두 합한다. 편향을 더합니다.  편향이 없으면, 추세선은 원점을 꼭 지나야 합니다. 활성 함수를 적용 해 비선형 함수로 만듭니다. 두 벡터의 내적은 다음과 같이 표현할 수 있습니다. 두 벡터의 내적 FC(Fully Connected) Layer Matrix 곱셈 연산 표현 방법 뉴런이 모인 한 단위를 계층(Layer)라고 하며, 이전 계층과 다음 계층의 모든 뉴런이 서로 연결된 계층을 Fully-Connected Layer(Dense Layer)라고 합니다. N개의 입력, M개의 출력이 있는 네트워크 예제입니다. 매트릭스  W 의  w 0 는 (N*1)의 벡터이며, 이런  w 0 를 M개 나열되어 있습니다. 입력  x 는 N개라 행렬로 표현하면 (N*1)로 표현됩니다. 가중치를 transpose하여 (M*N)*(N*1)을 연산하여 출력은 (M*1) 형태가 됩니다. 얕은 신경망 - Shallow Neural Network 구조 얕은 신경망 - Shallow Neural Network 입력, 은닉, 출력 3개의 계층으로 되어 있으며, 은닉 계측과 출력 계층이 Fully Connected 계층인 모델을 얕은 신경망(Shallow Neural Network)라고 합니다. 입력 계층(Input Layer) 아무런 연산 없이 은닉계층으로 값을 전달함. 계층의 크기=Node의 개수=입력 Scalar의 수=입력 Vecto...

SentencePiece 설치 및 사용법

Jupyter notebook 사용 기준으로 설치 및 사용법을 설명합니다. Sentencepiece 특징 및 기능 설명은 다음 사이트에서 참고 바랍니다. https://sungwoony.blogspot.com/2020/04/sentencepiece.html one-sentence-per-line raw corpus file.  tokenizer, normalizer 또는 preprocessor를 실행할 필요가 없습니다. Default로, Unicode NFKC로 SentencePiece input을 정규화 합니다.  설치방법 VM환경에 pip library가 설치되어 있어야 하며, 다음 명령어를 수행하면 설치됩니다. pip install sentencepiece 사용법 Sentencepiece library import setencepiece를 import해야 하며 관례적으로 spm으로 사용합니다. Train SentencePiece Model from corpuse botchan.txt 파일을 다음 사이트에서 다운로드 받을 수 있습니다. https://github.com/google/sentencepiece/blob/master/data/botchan.txt --mode_type의 기본은 uni type입니다. --model_type에 bpe를 넣으면 bpe 타입으로 생성됩니다. Train이 완료되면 "m.uni.model"과 "m.bpe.model" 이 생성됩니다. 둘 의 차이점은 다음에 확인할 수 있습니다. Option Name Description input 파일목록은 쉼표로 구분합니다. model_prefix output model로 prefix. <model_name>.model과 <model_name>.vocab 이 생성됩니다. vocab_size vocabulary size, e.g.. 8000, 16000, o...

[Deep Learning-딥러닝]SRU(Simple Recurrent Unit)

SRU(Simple Recurrent Unit) SRU는 병렬화와 시퀀스 모델링이 가능한 light recurrence한 unit입니다. 기존 RNN Architecture(RNN, LSTM, GRU)는 previous time step에 대한 의존성 때문에 병렬처리가 불가능하여 학습 속도가 느렸습니다. 이런 단점을 SRU는 높은 병렬화 및 시퀀스 모델링 기능을 제공하여 학습시간 단축시켰습니다. SRU는 분류, 질의응답에서 cuDNN-optimized LSTM보다 5~9배 빠른 속도를 보였고, LSTM과 convolutional models보다 좋은 성능을 보였습니다. 특징 SRU의 state연산은 time-dependent이지만, 각 state 차원은 independent입니다. 이것은 hidden dimension과 time steps에서 병렬화 연산하는 CUDA-level optimization으로 병렬 처리 가능합니다. SRU는 convolutions을 더 많은 recurrent 연결로 대체하였습니다(QRNN과 KNN과 같이). 이건 적은 연산으로 모델링을 유지합니다. SRU는 highway connection방법과 deep architecture에서 gradient전파에 맞게 조정된 매개 변수 초기화 체계를 사용하여 deep recurrent models training을 개선합니다. 연산방법 SRU는 forget gate, state, reset gate, hidden state 연산을 수행합니다. Light recurrence (1, 2번 수식) Forget gate: 정보 흐름을 제어 State: 이전 state와 현재 입력을 forget gate에 따라 조정 특징 - 문자열 정보를 위해 input x와 state c를 연산 - 이전 state에 영향을 받음 - 병렬처리를 위해 matrix multiplication 대신 point-wise multiplication 연산 수행 Highway network (3, 4번 수식) Hidden ...