Linux환경에서 Mmdetection2.X 버전 설치방법

원래는 워드에다가 연구실에서 진행하는 mmdetection 일지를 문서화하고 있었는데 

블로그에다가는 이를 정리해서  mmdetection에 대해  올리려고 한다.

 

mmdetection을 리눅스 서버에 2.x 버전으로 설치해본다.

3.X 버전은 이미 다른 서버에서 돌리고 있고, 관련 자료도 2.X 버전이 많기 때문에

2.X 버전으로 할려고 한다.

 

3.X버전은 공식문서를 참고해서 다운로드 하면 된다.

 

우선 리눅스 서버에 xrdp를 설치했다.

원격에서 접속해서 GUI 환경을 쓰는 것이 이미지 객체 인식에서 중요하다고 생각하기 때문이다.

 

리눅스 환경에서 mmdetection 설치 하는 법

아예 깡통 리눅스 서버에다가 설치하는 방법에 대해서 알아보자.

기본적으로 리눅스지식과 개발 경험이 있다고 가정하겠다.

우선 mmdetection을 사용하려면  

 

1.conda 설치

가상환경 conda가 있다면 파이썬 환경을 매우 손쉽게 다룰 수 있다.

현재 기준으로 anaconda는 홈페이지에 들어간 다음에 다운로드한 이후 bash로 설치할수 있다.

conda를 설치했다면

conda create --name openmmlab python=3.8 -y

위 명령어를 통해 가상환경을 만든다. (이름은 자유)

파이썬을 딱히 설치안해도 알아서 3.8버전을 깔아준다. 

conda activate openmlab

이 명령어는 생성한 가상환경에 접속하는 명령어다.

가상환경이 있으면 해당 가상환경에 환경별로 여러가지 라이브러리들을 설치할수가 있어서 간편하게 바꿔서 쓸수가 있다.

(예를 들면 핸드폰이 여러개인거죠 업무용 갤럭시 핸드폰, 사진촬영용 아이폰 처럼 몇개 가지고 있는 겁니다)

 

2.git 설치

sudo apt-get install git

Git을 설치해야 github에 있는 mmdeteciton을 다운받을 수 있다.

 

3.아래 순서대로 pytordch와 mmcv를 설치한다.

conda install pytorch==1.13.0 torchvision==0.14.0 pytorch-cuda=11.6 -c pytorch -c nvidia
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu116/torch1.13/index.html

 

여기서 conda install ~~을 통해서 CUDA도 같이 설치할 수 있다.

CUDA는 엔비디아에서 GPU 작업을 편하게 만들어주는 툴이다.

4.이후 mmdetection을 설치한다.

git clone --branch 2.x https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .

 

5.mim, openmmin, mmengine 설치

pip install -U openmim
mim install mmengine
pip install openmim

 

이로서 설치가 완료 되었다.

이후에는 mmdetection을 간단히 테스트 하는 방법에 대해 알아본다.