libtorch yolov3

libtorch-yolov3

순수 C++로 작성된 YOLO v3 객체 감지 알고리즘의 Libtorch 구현입니다. 빠르고 쉽게 프로덕션에 통합할 수 있으며 CPU와 GPU가 모두 지원됩니다. 즐기세요 ~

이 프로젝트는 pytorch 버전에서 영감을 받아 C++로 다시 작성했습니다.

요구사항

1. 리브토치 v1.0.0
2. 쿠다
3. OpenCV(예제에서만 사용됨)

컴파일하려면

1. cmake3
2. gcc 5.4 +

 mkdir build && cd build
cmake3 -DCMAKE_PREFIX_PATH="your libtorch path" ..

# if there are multi versions of gcc, then tell cmake which one your want to use, e.g.:
cmake3 -DCMAKE_PREFIX_PATH="your libtorch path" -DCMAKE_C_COMPILER=/usr/local/bin/gcc -DCMAKE_CXX_COMPILER=/usr/local/bin/g++ ..

감지기 실행

가장 먼저 해야 할 일은 v3의 가중치 파일을 가져오는 것입니다.

 cd models
wget https://pjreddie.com/media/files/yolov3.weights 

단일 이미지의 경우:

 ./yolo-app ../imgs/person.jpg

테스트한 결과 GPU(1080ti)에서는 25ms가 소요됩니다. 추론 작업을 두 번 이상 실행하고 평균 비용을 계산하세요.