yolov5

yolov5 windows 10 설치

영성 2021. 12. 27. 17:22

 

(참고)cuda version 10.1 기준으로 설치를 했습니다

 

 

Anaconda Prompt에서 가상환경을 생성하고 접속

(conda create -n 가상환경이름 python=파이썬버전)
(conda activate 가상환경이름)

conda create -n yolov5 python=3.8
conda activate yolov5

 

 

 

yolov5 설치하고 yolov5폴더로 이동

(cd yolov5를 설치하고자 하는 폴더위치)
(git clone 깃허브주소)

cd C:/
git clone https://github.com/ultralytics/yolov5.git
cd yolov5

빨간색 네모 안에 버튼을 누르면 주소가 복사된다

 

 

 

 cuda version을 확인

nvcc --version

- cuda가 설치되어 있지 않은 경우에는 밑에 주소에서 다운 받으시면 됩니다 (다운 받고나서는 anaconda prompt를 재부팅) 

- 저는 10.1을 사용하기에 10.1 버전기준으로 알려드리겠습니다

https://developer.nvidia.com/cuda-10.1-download-archive-base

 

 

 

설치 (cuda10.1)

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

!주의 : cudatoolkit은 반드시 내 cuda 버전과 같아야 합니다

 

 

 

필요한 라이브러리들을 설치

pip install -r requirements.txt

!주의 : 라이브러리는 yolov5와 나의 가상환경 설정에 따라 보이는 것과 버전은 다를 수도 있습니다

- 정말 드물게 라이브러리를 한번에 받다가 오류가 생기는 경우가 있는데 yolov5 tutorial을 실행할 때 오류가 생긴다면 먼저는 오류가 뜨는 라이브러리를 다시 설치해보고 그래도 안된다면 완전히 다 지웠다가 가상환경부터 다시 설치해보고 라이브러리들도 하나씩 설치해보면 해결되기도 합니다 

 

 

 

테스트

python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images

- yolov5폴더 안에 run/detect/exp 폴더안에 밑에 처럼 사진에 객체감지(object detection)가 된다면 성공입니다 

 

 

다음에는 custom data를 활용한 훈련과 결과에 대해서 소개드리겠습니다