본문 바로가기
jetson nano 실습

[ActionAI] 최종 실행

by 코딩새내기_ 2021. 7. 7.

ActionAI를 실행하기 위해서는 다른 오픈소스보다 더 환경설정을 잘 해야 실행할 수 있다.

첫 번째 문제점은 ActionAI에서 trt모델을 받아 사용하였는데 engine오류가 뜨면서 모델을 불러올 수 없었다.

ActionAI에서 trt모델을 다운 받아서 사용하면 안되고, 사용자가 직접 trt_pose저장소에서resnet18_baseline_att_224x224_A_epoch_249.pth모델을 받고 직접 tensorrt로 변환해주어야 한다.

거기에서 얻은 resnet18_baseline_att_224x224_A_epoch_249_trt.pth모델을 사용해야 했었다.

 

그리고 jetson nano에서 pytorch와 tensorflow를 동시에 설치해야 했었는데 최신버전에서는 서로 지원하는 numpy버전이 달라서 오류가 뜨게 된다. 그래서 pytorch==1.7.0, tensorflow==2.3.1, numpy==1.18.5버전을 사용해야지 버전 충돌을 막을 수 있다.

 

환경설정 때 크게 애 먹은 부분은 위 두가지 정도가 있었다.

 

https://ddo-code.tistory.com/21

https://ddo-code.tistory.com/23

https://ddo-code.tistory.com/25

를 순서대로 진행하여 실행할 수 있었다.

 

댓글