Linux(Ubuntu 20.04) 에 OpenPose를 이용해 영상 속 자세 추정하기

이전 글과 이어집니다.

Ubuntu 에서 C 혹은 Python을 돌리기 위해서는 VS Code를 이용하거나 터미널을 이용합니다.

VS Code에서는 윈도우에서 사용한 Visual Studio 와 비슷해 여러 라이브러리와 언어를 설치해 돌리면 됩니다.

오늘은 Terminal 에서 돌리는 방법을 적어보겠습니다.

 

우선 작업 전 항상 업데이트를 진행합니다. Ctrl + Alt + T를 눌러 터미널을 불러옵니다.

sudo apt update

터미널에 코드를 복사 붙여넣기 할때는 복사는 Ctrl+C를 하고 터미널에 붙여 넣기는 Ctrl+Shift+V를 눌러야 정상적으로 붙여넣기가 됩니다.

이후 OpenPose 학습을 위해 Python을 설치해야 하는데 Ubuntu 20.04는 Python이 자동으로 설치가 되어 있습니다. 다른 버전은 설치 하셔야 합니다.

파이썬의 버전을 확인합니다.

python3 --version

이후에는 필요한 모듈 패키지를 설치해야 합니다.

sudo apt-get install python3-pip

보통 필요한 모듈은 설치가 되지만 우리가 필요한 모듈은 설치 포함이 안된것이 많아 따로 명령어로 설치해줘야 합니다. 참고로 Ubuntu에서 이전 명령어를 다시 치려면 방향키 위를 누르면 이전 명령어가 다시 나타납니다.

sudo apt install python3-opencv

pip install tqdm

glob는 버전 2로 설치합니다

pip install glob2

패키지 설치가 끝났으면 이전 포스팅에서 다운 받은 파일을 Ubuntu 에서도 다운 받습니다.

https://github.com/CMU-Perceptual-Computing-Lab/openpose

GitHub - CMU-Perceptual-Computing-Lab/openpose: OpenPose: Real-time multi-person keypoint detection library for body, face, hand

다운 로드가 끝나면 압축 해제를 합니다.

그런 다음 models 파일로 들어가 오른쪽 마우스를 클릭해 models 파일에서 터미널을 엽니다.

getModels를 설치 하기 위해 sh 파일을 실행합니다. (Ubuntu 에서는 sh 파일이 설치실행 파일입니다)

./getModels.sh

설치되는것을 확인할 수 있습니다.

설치가 끝난 후 다시 바탕화면으로 돌아와 구글 코랩처럼 폴더을 하나 만듭니다. 저는 openpose를 만들었습니다.

cd ..

이전 폴더 경로로 이동하는 명령어 입니다.

mkdir openpose

openpose라는 폴더를 만드는 명령어 입니다.

사진과 같이 폴더가 만들어진것을 확인할 수 있습니다.

다음으로는 이전 포스팅과 동일하게 openpose폴더 안에 필요한 영상, 쓰고자 하는 pose 네트워크 파일을 넣어줍니다.

 

파일을 옮긴 후 학습을 돌리기 위해 파이썬 파일을 하나 만듭니다.

cd openpose

touch 는 파일 생성 명령어 입니다

touch pose.py

새롭게 탄생할 프레임을 저장할 폴더를 만들어 줍니다 (저는 footmove로 만들었습니다)

pose.py을 열어 아래의 코드를 제일 첫 줄에 입력합니다.

#1/usr/bin/env python

파일의 권한을 부여하고 실행하겠다는 명령어 입니다.

이후 코랩의 코드에서 각각 파일들과 저장 파일들의 경로를 바꿔준 뒤 나머지 코드는 동일합니다

우분투의 기본적인 파일 경로는 /home/자신의 username/자신이 만든 폴더명 입니다

저는 /home/leesohwa/openpose/파일명 이렇게 되겠죠?

 

이후 이전 게시물의 코드를 복붙한 뒤 Ctrl+s를 눌러 저장하고 파일을 실행합니다.

 

이외에도 VSCode에서 작동 방법도 동일합니다.