[CV] Detectron2 모델 환경 설정

detectron2 모델 환경 설정

1. ubuntu 환경 확인 

(base) ➜  ~ lsb_release -a

 

2. python 버전 3.8로 detectron_env 이름을 가진 가상환경을 하나 만들어준다. 

(base) ➜ conda create -n detectron_env python=3.8
(base) ➜ conda activate detectron_env # 가상환경 활성화

 

3. detectron2 github에서 git clone을 실행하고 폴더 경로로 들어간다.

(detectron_env) ➜ git clone https://github.com/facebookresearch/detectron2.git
(detectron_env) ➜ cd detectron2

 

4. detectron_env 가상환경 내 pytorch 설치한다.

• "detectron2/ " 는 3번에서 cd detectron2 명령어를 통해 해당 폴더로 들어온 것을 의미한다.

(detectron_env) ➜ detectron2/ conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

 

5. detectron2 관련 모듈을 한번에 설치한다.

• detectron2 폴더로 들어온 상태에서 진행해야한다.
• pip의 -e 옵션은 editable 옵션으로, setuptools의 develop mode와 동일하다.
  • 옵션을 사용했을 경우, egg-info 파일이 프로젝트 루트에 상대적으로 생성됨.

(detectron_env) ➜ detectron2/ pip install e .

 

6. opencv-python 설치한다.

(detectron_env) ➜ detectron2/ pip install opencv-python

 

참고 )

detectron2 github : https://github.com/facebookresearch/detectron2

pytorch previous-versions : https://pytorch.org/get-started/previous-versions/