HigherHRNet : Model 구현

지난번에 리뷰하였던 HigherHRNet을 드디어 돌리는데 성공하였다..

실패 및 성공할 수 있었던 Point는 다음과 같다.

1. Python 버전 (3.9.X는 Error 발생.. 3.6.8 에서 실행했더니 정상적으로 작동)

2. Cuda 버전 , Torch 버전 맞추기 --> 이 부분이 제일 난감하고.. 몇번이나 했는지

3. Conda 가상환경마다 다르게 적용되는 것이 무엇인지 확인하기 ( Cuda 또한 가상환경마다 다르다)

4. COCO Dataset의 방대한 양

5. COCO API 와 CrowdPose API 설정

결론적으로 Python = 3.6.X , Cuda = 11.6 , Pytorch = 해당 Cuda에 맞는 torch 이 핵심.

버전이 높은게 좋은게 아니고 버전 낮추는게 좋은듯...

 

최종 실행했을때 화면!

해당 화면 위에는 각 layer들이 설명되어있음. 

요런식으로... 

그래서 결과물은 어떻게 나오나? 

성공적인 결과물

이외에 정상적으로 처리 안되었다고 생각하는 결과물

 

해당 사진들이 결과물로 나왔음.. 

다음 목표는 직접 Input하여 Output을 얻는 것, 각 좌표분석!

성공하니 뿌듯 :)