[YOLO] windows10 에서 YOLO 실행하기 (cmake)

최재혁·2022년 3월 1일
0

YOLO refinement

목록 보기
2/3

cuda 11.5
cudnn v8.3.1 for CUDA 11.5
opencv 3.4.16
cmake cmake-3.23.0-rc2-windows-x86_64
visual studio 2019 (언어 확장팩 영어 설치)

visual studio 2019


cuda 설치

https://developer.nvidia.com/cuda-11-5-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

cudnn 설치

https://developer.nvidia.com/rdp/cudnn-archive

zip 파일 압축 해제 후 폴더
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5
에 붙여넣기

opencv 설치

https://opencv.org/releases/

시스템 환경변수 path 추가 opencv\build... 아래 6개

cmake 설치 및 visual studio 연동


설치 중간 path 설정하기 current user로 해도 됨

CMake-gui 실행


1. where is the source code
darknet 경로 설정
2. where to build the binaries
darknet 경로 설정

3. configure 누른 다음 visual studio 2019 선택 후 확인

4. generate

5. open project
Release로 설정

솔루션 빌드

Darknet 실행

1. file 복사

\darknet\3rdparty\pthreads\bin\pthreadVC2.dll
\darknet\Release\darkent.exe

2개 파일 \darknet 폴더에 복사

2. weight file 및 demo 실행
weight file (yolov4.weight)
https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights

다운로드 후 \darknet 폴더에 넣기

#darknet 경로 이동후

darknet.exe detect cfg/yolov4.cfg yolov4.weights data/dog.jpg

첫 실행시 ERROR
TensorFlow Could not load library cudnn_cnn_infer64_8.dll. Error code 126 에러 발생

해결법
https://afsdzvcx123.tistory.com/entry/%EC%9D%B8%EA%B3%B5%EC%A7%80%EB%8A%A5-TensorFlow-Could-not-load-library-cudnncnninfer648dll-Error-code-126-%EC%97%90%EB%9F%AC-%ED%95%B4%EA%B2%B0#google_vignette

결과 화면

profile
Autonomous driving vision

0개의 댓글