openpose_wrapper

OpenPose のラッパーモジュール。
以下の機能を追加しています。

Docker を用いて OpenPose のインストールから実行までを自動的に行う。
サーバー機能を追加。

現在 GPU 版のみ動作します。

■ 目次

動作環境
使用法
1. サーバー機能を使用しない場合
  1. GPU 版 OpenPose
  2. CPU 版 OpenPose
2. サーバー機能を使用する場合
  1. GPU 版 OpenPose
  2. CPU 版 OpenPose
【参考】 OpenPose の主なオプション引数

■ 動作環境

Ubuntu :
docker, docker-compose
GPU 版 OpenPose
- nvidia 製 GPU 搭載マシン
- CUDA 10.1
- nvidia-docker2
リクエスト処理 : openpose_server/request.py
- tqdm
- requests
- Pillow

■ 使用法

◎ サーバー機能を使用しない場合

☆ GPU 版 OpenPose

「Ubuntu + GPU 版 OpenPose」の Docker イメージを使用して、コンテナ内で OpenPose を実行します。Docker イメージの作成には長時間がかかります。

Docker イメージの作成＆ OpenPose の実行
以下のコマンドを実行。

$ sh run_openpose_gpu.sh

又は、以下のコマンド例を実行。

$ docker-compose -f docker-compose_gpu.yml up -d
$ docker exec -it -u $(id -u $USER):$(id -g $USER) openpose_ubuntu_gpu_container bash

# コンテナ内 : /home/ubuntu/share/openpose_wrapper
$ cd openpose_gpu

# コンテナ内 : /home/ubuntu/share/openpose_wrapper/openpose_gpu で OpenPose 実行
$ ./build/examples/openpose/openpose.bin \
    --model_pose COCO \
    --image_dir ../sample_n5 --write_json ../results_json --write_images ../results_image \
    --display 0 \
    --hand

☆ CPU 版 OpenPose

準備中...

◎ サーバー機能を使用する場合

☆ GPU 版 OpenPose サーバー

サーバー機能非使用時の GPU 版 Docker イメージと同様のイメージを使用し、コンテナ内でサーバーを起動して OpenPose を実行します。
サーバー機能使用時は、デフォルト設定では、5010 番ポートが開放されている必要があります。使用するポート番号は、docker-compose_gpu.yml 内の ports: タグ、及び、openpose_server/app.py, openpose_server/request.py の --port 引数の値を設定することで変更できます。

Docker イメージの作成＆コンテナの起動＆リクエスト処理
以下のコマンドを実行。

$ sh run_api_gpu.sh

又は、以下のコマンドを実行

# API の起動
$ docker-compose -f docker-compose_gpu.yml stop
$ docker-compose -f docker-compose_gpu.yml up -d

# OpenPose サーバーへのリクエスト処理
$ cd openpose_server
$ python request.py \
    --host 0.0.0.0 --port 5010 \
    --image_dir ../sample_n5 \
    --write_json ../results_json

☆ CPU 版 OpenPose サーバー