본문 바로가기

팀프로젝트 - TailsRoute

TEAM Project - 모델훈련 한계

(1) Keypoint 추출 (Pose Estimation)

  • HRNet 또는 KeypointRCNN 사용 : 프레임 / 15개의 keypoint 좌표 (x,y)

 

3시간이 넘어도... 진행이 되지 않아서 HRNet을 사용하는 부분은 패스......
(좋은 그래픽 카드 없어서.. 🥲) 한계.다........

HRNet 모델이 매우 복잡하고, CPU만으로 실행했기 때문...

HRNet은 고해상도의 입력 데이터를 처리하는 고급 모델로, CPU에서의 계산량이 많음..

좀 더 정확도를 높이기 위해 키포인트를 재추출하여 정확도를 향상시키려고 도입하였는데.. 지금으로서는 시간 부족으로 모델 훈련 후 시간이 남는다면 batch 처리하여 진행
(batch 처리로 여러 이미지를 한 번에 처리하면 속도가 개선될 수 있도록)