ValueError: Cannot load file containing pickled data when allow_pickle=False 에러 해결

HeyHo·2022년 8월 30일
0

Pytorch를 통해서 모델을 생성하고 트레이닝 하는 과정에서, 다음과 같은 에러가 발생하였다. 데이터 로더 쪽에서 문제가 발생하였다.

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/park/anaconda3/lib/python3.7/site-packages/numpy/lib/npyio.py", line 445, in load
    raise ValueError("Cannot load file containing pickled data "
ValueError: Cannot load file containing pickled data when allow_pickle=False

다음과 같은 문제가 발생한 이유는, np.load(datadir) 메소드를 사용하는 과정에서 png파일을 np.load로 바로 사용하여서이다.
단순하게 이미지를 load 해보면 다음과 똑같이 문제가 발생하는 것을 알 수 있다.
np.load를 사용하기 전에 png 파일을 numpy 파일로 변환해주면 문제가 해결될 것이다.

profile
Coputer vision, AI

0개의 댓글