MediaPipeで姿勢推定を行い、Tokyo2020オリンピック風のピクトグラムを表示するデモ

Overview

Tokyo2020-Pictogram-using-MediaPipe

MediaPipeで姿勢推定を行い、Tokyo2020オリンピック風のピクトグラムを表示するデモです。

Tokyo2020Pictgram02.mp4

Requirement

  • mediapipe 0.8.6 or later
  • OpenCV 3.4.2 or later

Demo

以下コマンドでデモを起動してください。
ESCキー押下でプログラム終了します。

python main.py
  • --device
    カメラデバイス番号の指定
    デフォルト:0
  • --width
    カメラキャプチャ時の横幅
    デフォルト:640
  • --height
    カメラキャプチャ時の縦幅
    デフォルト:360
  • --static_image_mode
    静止画モード
    デフォルト:指定なし
  • --model_complexity
    モデルの複雑度(0:Lite 1:Full 2:Heavy)
    ※性能差はPose Estimation Qualityを参照ください
    デフォルト:1
  • --min_detection_confidence
    検出信頼値の閾値
    デフォルト:0.5
  • --min_tracking_confidence
    トラッキング信頼値の閾値
    デフォルト:0.5
  • --rev_color
    背景色とピクトグラムの色を反転する
    デフォルト:指定なし

Using Docker

Ubuntuの場合はホストマシンにMediaPipeをインストールせず、Docker + docker-composeを使うこともできます。

まず環境に合わせてdocker-compose.ymlを編集します。
ビデオデバイスを指定する際video0を使う場合は以下のように編集します。

    # Edit here
    devices:
      # - "/dev/video0:/dev/video0"
      # - "/dev/video1:/dev/video0"
-     - "/dev/video2:/dev/video0"
+     - "/dev/video0:/dev/video0"

次にDockerイメージをビルドします。

docker-compose build

最後にDockerコンテナを起動します。

docker-compose up

Author

高橋かずひと(https://twitter.com/KzhtTkhs)

License

Tokyo2020-Pictogram-using-MediaPipe is under Apache-2.0 License.

You might also like...
Comments
  • Add Dockerfile

    Add Dockerfile

    Ubuntu用にDockerfileおよびdocker-composeを追加します。

    GPU有無の影響を受けずにDockerコンテナを起動できるように環境変数は少し工夫しています。

    以下の環境で動作確認しました。

    • Ubuntu 18.04 with GPU
      • CPU: Intel Core i5-9600K
      • GPU: RTX2070 super
    • Ubuntu 18.04 with Intel Integrated GPU
      • CPU: Intel Core i7-8650U
    opened by Tiryoh 1
Owner
KazuhitoTakahashi
KazuhitoTakahashi