Windows 에서 Python YoloV5 & Cuda 준비

일단 나는 개발을 윈도우 환경에서 하므로....

윈도우 환경에 YoloV5 를 진행해보려고 한다.

 

일단 구글에서 검색해보면... 대부분 아나콘다 를 사용하여 가상의 환경에서 YoloV5 를 구축하던데.

나같은경우에는 가상환경을 추천하지 않는다..

 

(물론.. 공부하는 입장에서 언제든지 환경을 새로 구축할수 있다는 장점이 있지만 타 언어와의 연동 및 산업현장 PC에 적용하기엔... 가상환경이 그리 좋은 대안은 아니다.)

 

일단 가장먼저 해야할일은 yolov5 를 다운받아야한다.

https://github.com/ultralytics/yolov5

GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

 

대충 원하는 폴더에 다운받아서 압축을 풀자..

 

일단 내 개발환경은

CPU : Intel i9 GPU : RTX 3080 Python : 3.7 Cuda : 11.4.120 Cudnn : cudnn-11.4-windows-x64-v8.2.2.26

 

내가 다운받은 경로는

D:\yolov5-master\yolov5\

압축을 풀어보면 아래와 같이 나온다.

그럼 일단 requirements.txt 파일을 메모장으로 열자

위 내용처럼.. 필요한 것들을 설치해주자... 

파일을 한번에 읽어서 설치하는 방법도 있긴한데 비추.. 에러날 확율이 있음.

일단 이대로 설치를 해주면.... YoloV5 가 CPU 로 동작한다.

 

https://pytorch.org/get-started/locally/

PyTorch

개발환경에 맞춰 선택을 해준다음..

위에 나오는 커맨드를 실행해주면 된다..

나는 CUDA 11.4 인데도.. 그냥 11.3 선택해서 입력하니까 잘됨..

 

요로케.. 나오면 성공~

 

이제 아까 yolov5 압축푼 폴더로 이동해서 아래와 같이 실행하자.

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

그럼 요로케 위와 같이 CUDA 를 통해 결과 가 나오면 성공~!

그럼 runs\detect\exp11 폴더로 생성됬다고 하니 들어가서 보면

요러케 2개의 이미지를 확인할수 있다..

 

그럼 YoloV5 설치 끝.

이제 YoloV5 를 내가 원하는대로 학습하고 인식해보고 최종적으로 C# 과 연동하는것까지 글을 작성해보려 한다.