[Windows 10 + Detectron2] Facebook(Meta AI) Detectron2 설치

Facebook(Meta AI)의 Detectron2를 윈도우 환경에서 설치해보자!

 

1. Anaconda Prompt 에서 Python 버전 3.7의 가상환경을 생성

conda activate -n 가상환경이름 python=3.7

아래 사진은 생성한 가상환경을 activate 한 후 설치된 python 버전을 확인한 모습이다.

 

2. PyTorch 설치하기

Detectron2의 공식문서를 확인하면 아래와 같이 PyTorch는 1.8 이상을 요구한다고 한다. (자세한 설치 방법은 여기 참조)

 

PyTorch 공식사이트를 참고해서 아래와 같이 설치했다.

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

 

3. Cython 과 Pycocotools 설치

pip install cython
pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"

 

4. detectron2 윈도우 버전 설치

git clone https://github.com/DGMaxime/detectron2-windows.git

 

5. 다운받은 detectron2 환경 세팅하고 opencv 설치

cd detectron2-windows 
pip install -e .
pip install opencv-python

 

6. tests 폴더로 이동하여 test 코드 실행

cd tests
python test_windows_install.py

아래와 같은 결과가 뜨면 detectron2 설치 완료!

 

 

Reference

https://medium.com/@yogeshkumarpilli/how-to-install-detectron2-on-windows-10-or-11-2021-aug-with-the-latest-build-v0-5-c7333909676f

How to Install Detectron2 on Windows 10 or 11 –2021(AUG) with the latest build(v0.5).