Jetson Nano YOLOv5 설치 (Intro)

이도형·2024년 5월 10일
1
post-thumbnail

YOLO 딥러닝 모델을 Jetson Nano 기기에 활용하기 위해
많은 글을 찾아봤지만 버전 문제로 오류가 나는 경우가 많습니다.😭

하지만 의지의 한국인의 자세로.... 해!결!
간단한 팁을 공유해드리고자 합니다 (물론 저도 찾아봐야되서)


설치 방법

YOLOv5를 실행시키기 전에 꼭 설치해야될 것이 있습니다.

1. 버전에 맞는 Opencv 설치 하기!

2. 버전에 맞는 Pytorch 설치 하기!

3. 버전에 맞는 Torchvision 설치 하기!

4. YOLOv5 Git Clone 하기 (중요!!!)


YOLO Inference 속도 향상

YOLOv5를 설치하고 Inference를 진행하면 매우매우매우 느리다는 것을 알 수 있습니다.
이것을 개선하기 위해서도 몇 가지 방법이 있습니다.

YOLO 학습 시 Image Size 줄이기

Tensor RT 활용하기

Fan (발열 방지) 사용하기

최대 전력 사용하기


결과

10fps까지 나오지만 조금 더 개선이 필요!


마치며

부득이하게 글을 방법 별로 나누게 되었습니다..😅
(그래야 더 찾기 편하지 않을까요..??)
속도 향상 혹은 기타 도움 되는 정보는 계속 수정할 예정입니다!

profile
열심히 살고 싶습니다! 일하고 싶습니다 :)

0개의 댓글

관련 채용 정보