MMDetection (1)

Chapter0 MMDectection 소개

• 중국의 openMMlab을 중심으로 만든 Pytorch기반의 오픈소스 패키지
• 다양한 object detection, instance segmentation, panoptic seqmentation 알고리즘을 패키지로 제공

Chapter1 MMDectection 설치하기

• CUDA 설치
  -> 장치관리자에서 GPU모델 확인
  -> GPU 성능 확인: 링크를 통해 GPU 모델 번호 확인
  -> GPU 성능이 CUDA 11.8버전을 이용할 수 있는지 확인 (성능이 3.5~9.0이면 가능)
  

-> CUDA 다운로드 후 설치 (CUDA 11.8 설치 링크)

-> cuDNN 11.X 버전 설치
-> 환경변수 편집 (편집 후 재부팅)
-> Miniconda 설치

1) 가상 환경 생성

• Anaconda Prompt를 실행

conda create -n mmdetection python=3.11

conda activate mmdetection

2) PyTorch 설치

• CUDA 버전에 맞는 PyTorch 설치

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

3) openmim 설치

pip install -U openmim

4) mmcv 설치

pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html

5) mmengine 설치

mim install mmengine

6) MMDetection 설치

• GitHub에서 MMDetection 저장소 클론

git clone https://github.com/open-mmlab/mmdetection.git

cd mmdetection

• 의존성 패키지 설치

pip install -r requirements/build.txt

-> 설치가 되지 않을 경우

pip install cython

pip install numpy

• MMDetetcion 설치

pip install -v -e .

7) 설치 확인

이 글은 제로베이스 데이터 취업 스쿨의 강의 자료 일부를 발췌하여 작성되었습니다