No module named 'albumentations'

boingboing·2024년 3월 12일

원인

  • albumentations 라는 이미지 augmentation 모듈이 없어서 발생하는 에러코드이다.

해결 방법

  • 간단하게 albumentations 를 pip 으로 설치해주면 해결된다.
pip install albumentations

-> 이걸 설치하는 과정 중 dependency 중에 opencv-python-headless를 설치하는 과정이 있음.

그런데 이 과정에서...!

현상1

Building whell for opencv-python-headless(pyproject.toml) 여기서 멈춤....

기다려보니..

현상2

  Exception: Not found: 'python/cv2/py.typed'
  ----------------------------------------
  ERROR: Failed building wheel for opencv-python-headless

sol1

  • Python 3.6 버전에 대응하는 OpenCV 버전은 4.6인데, yaml 파일에 그 버전을 명시하지 않으면 못 찾는듯. (지금 그냥 깔면 opencv 4.9설치함)

    The last OpenCV version to have wheels built for Python 3.6 was v4.6. With that in mind, I was able to get everything installed by explicitly restricting to that version.

  • torch 1.0.0 쓸 거 아니면 굳이 3.6 쓸 필요X -> 3.8로.
  • 3.8로 한 이유: torch 1.5는 -> 3.5, 3.6, 3.7, 3.8, and 3.9 중 하나 쓰면 되는데 3.6이 opencv설치 에러나니까.

-> conda 환경에서 python 3.8로 재설치 후 다시 opencv 깔아 봄.

-> 해결됨. pip install opencv-python-head 명령어 실행 시 python 버전에 따라 자동으로 다운로드하는 opencv가 4.9인데, 파이썬 3.6과 호환이 안 되는 거였음.

해결

-> python 버전을 바꾸니 해결 됨.

0개의 댓글