Darknet(yolo)모델을 TensorRT로 변환하기

그래픽카드가 크게 발전하면서 인공지능은 탄탄대로를 걷고 있다. 심지어 NVIDIA에서 Xavier계열의 보드를 사용할 때, TensorRT 모델을 지원해주기 때문에 임베디드에도 인공지능 모델을 쉽게 최적화하여 넣을 수 있다. 이번 글에서는 다크넷 기반의 Yolo모델들을 TensorRT로 변환하는 법에 대해서 공부해보고자 한다. **참고로 필자는 NVIDIA AGX Xavier 환경에서 실행하였다. 사실 말이 거창하지 이미 Converting Tool은 능력자님들께서 github에 올려놓으셨다. 이 글은 다음 github를 참고하면서 만들었다. github.com/jkjung-avt/tensorrt_demos jkjung-avt/tensorrt_demos TensorRT YOLOv4, YOLOv3, SSD..

2021.05.07