PyTorch의 Keypoint R-CNN으로 실시간 Skeleton Detecting

생각보다 PyTorch로 실시간 골격 검출에 관한 자료가 거의 없고 Github에도 코드가 내 것 포함 2개 밖에 없던데, 다 Matplotlib만 써서 그렇다.

전부 정적인 코드라서 실시간으로 하는 코드가 없다.

OpenCV 버전이 없어서 처음 만들 때 엄청 고생해서 만들었다.

 

Pytorch에는 기본적으로 성능 좋은 몇 가지 모델이 제공되는데, 그 중 하나가 keypointrcnn_resnet50_fpn.pth이다.

 

PyTorch는 CUDA버전과 CPU버전이 있는데 CUDA 쪽이 당연하게도 성능이 훨씬 좋다.

체감상 약 20프레임 정도 차이가 난다.

 

구동 테스트 환경은 

 

Opecnv 3.4.2

Torch 1.7.0+cu110

TorchVision 0.8.1+cu110

 

이다.

 

코드는

 

github.com/elppa1200/pytorch_skeleton_detection

elppa1200/pytorch_skeleton_detection