오늘은 학교 수업시간에 YOLO5를 사용해본 것을 글로 남기겠다.

초기 설정

일단 간단하게 설정과정을 요약하면,
1. conda설치
2. 가상환경 생성
3. pytorch install
4. CUDA 12.1 설치
5. cuDNN v8.9.7

1과 2는 저번에 설명을 하였기에 생략을하고 3,4,5를 설명하겠다.

pytorch install

https://pytorch.org/get-started/locally/이 사이트로 접속하여

사진과 같이 세팅한 후 아래 나오는 명령어로 install을 받아주면 된다.

CUDA

https://developer.nvidia.com/cuda-12-1-0-download-archive?target_os=Windows이 링크로 들어가서 CUDA를 다운받고 설정해주면 된다.

cuDNN

https://developer.nvidia.com/rdp/cudnn-archive이 링크의 아카이브로 들어가서 CUDA버전에 맞는 걸 다운받아주면 된다.

그 후에 cuDNN의 압축을 풀고 나온 파일을 CUDA의 경로에 들어가 복붙해준다.

그럼 끝이다.

detect.py를 사용해보기

python detect.py --source https://www.youtube.com/watch?v=jr9QMB9Sc5E
이 명령어를 사용해서 detect.py를 사용해서 유튜브 영상을 사용해서 인식시켜보자.

이런식으로 인식을 해준다.