AI（人工知能） 2020.12.28 2021.05.27 cedro

MediaPipeで、顔・手・ポーズのリアルタイム検出を行う

1.はじめに

　ちょっと前まで、顔・手・ポーズのリアルタイム検出をしようと思ったら高性能なGPUを積んだPCが必要でしたが、今回はこれらをCPUだけでも行える MediaPipe をご紹介します。

2.MediaPipeとは？

　MediaPipieは、Googleが提供しているライブメディアやストリーミングメディア向けのMLソリューションで、以下の特徴があります。

高速なエンドツーエンド処理：一般的なハードウェアでも高速化されたML推論と処理を実現
様々なプラットフォームで使用可能：Android、iOS、C++、Pythonに対応
無料のオープンソース： Apache 2.0で、拡張やカスタマイズが可能

　現在、4つのプラットフォームで、15種類のソリューションを提供されています。今回はその中から、Pythonプラットフォームの、顔検出、手検出、ポーズ検出、ホリスティック検出（顔・手・ポーズの統合検出）についてご紹介します。

3.コード（動画から検出）

　最初のデモはオジサンより、女優やダンサーの方が見ていて楽しいと思うので、あえて動画からホリスティック検出（顔・手・ポーズの統合検出）をやってみます。

　コードはGoogle Colabで動かす形にしてGithubに上げてありますので、それに沿って説明して行きます。自分で動かしてみたい方は、この「リンク」をクリックし表示されたノートブックの先頭にある「Colab on Web」ボタンをクリックすると動かせます。なお、GPUは使っていません。

　まず、セットアップを行います。

# ライブラリーのインストール
!pip install numpy==1.19.3
!pip install mediapipe

# github からコードをコピー
!git clone https://github.com/cedro3/mediapipe.git
%cd mediapipe/

# ライブラリーのインストール

!pip install numpy==1.19.3

!pip install mediapipe

# github からコードをコピー

!git clone https://github.com/cedro3/mediapipe.git

%cd mediapipe/

# 初期設定
import mediapipe as mp
mp_holistic = mp.solutions.holistic

# Initialize MediaPipe Holistic.
holistic = mp_holistic.Holistic(
    static_image_mode=True, min_detection_confidence=0.5)

# Prepare DrawingSpec for drawing the face landmarks later.
mp_drawing = mp.solutions.drawing_utils 
drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)

# 初期設定

import mediapipe as mp

mp_holistic = mp.solutions.holistic

# Initialize MediaPipe Holistic.

holistic = mp_holistic.Holistic(

static_image_mode=True, min_detection_confidence=0.5)

# Prepare DrawingSpec for drawing the face landmarks later.

mp_drawing = mp.solutions.drawing_utils

drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)

　次に、サンプル動画を読み込み静止画に分解し、imagesフォルダーに保管します。既にimages フォルダーがある場合（2回目以降）は、既にあるフォルダーを一旦削除してから処理を開始します。

　ご自分の動画を使いたい場合は、その動画をPCから./mediappipe/video にドラッグ＆ドロップでアップロードし、video_path = ‘./video/＊＊＊.mp4 の＊＊＊.mp4 をご自分で用意されたファイル名に変更すればOKです。

# ビデオの指定
video_path = './video/satomi.mp4'

1 2	# ビデオの指定 video_path = './video/satomi.mp4'

# video2images
import os
import shutil
import cv2

# 既にimagesフォルダーがあれば削除
if os.path.isdir('images'):
    shutil.rmtree('images')

os.makedirs('images', exist_ok=True)
 
def video_2_images(video_file= video_path,   
                   image_dir='./images/', 
                   image_file='%s.png'):
 
    # Initial setting
    i = 0
    interval = 3
    length = 300
    
    cap = cv2.VideoCapture(video_file)
    while(cap.isOpened()):
        flag, frame = cap.read()  
        if flag == False:  
                break
        if i == length*interval:
                break
        if i % interval == 0:    
           cv2.imwrite(image_dir+image_file % str(int(i/interval)).zfill(6), frame)
        i += 1 
    cap.release()  
 
def main():
    video_2_images()

# video2images

import os

import shutil

import cv2

# 既にimagesフォルダーがあれば削除

if os.path.isdir('images'):

shutil.rmtree('images')

os.makedirs('images', exist_ok=True)

def video_2_images(video_file= video_path,

image_dir='./images/',

image_file='%s.png'):

# Initial setting

i = 0

interval = 3

length = 300

cap = cv2.VideoCapture(video_file)

while(cap.isOpened()):

flag, frame = cap.read()

if flag == False:

break

if i == length*interval:

break

if i % interval == 0:

cv2.imwrite(image_dir+image_file % str(int(i/interval)).zfill(6), frame)

i += 1

cap.release()

def main():

video_2_images()

　そして、MediaPipeを使って、images フォルダーの中にある静止画から顔・手・ポーズを検出し、結果を上書きして行きます。

import cv2
from google.colab.patches import cv2_imshow
import numpy as np
import glob

# image file names to files in list format
files=[]
for name in sorted(glob.glob('./images/*.png')):
    files.append(name)

# Read images with OpenCV.
images = {name: cv2.imread(name) for name in files}

for name, image in images.items():
  # Convert the BGR image to RGB and process it with MediaPipe Pose.
  results = holistic.process(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))

  # Draw pose landmarks.
  annotated_image = image.copy()
  mp_drawing.draw_landmarks(annotated_image, results.left_hand_landmarks, mp_holistic.HAND_CONNECTIONS)
  mp_drawing.draw_landmarks(annotated_image, results.right_hand_landmarks, mp_holistic.HAND_CONNECTIONS)
  mp_drawing.draw_landmarks(
      image=annotated_image, 
      landmark_list=results.face_landmarks, 
      connections=mp_holistic.FACE_CONNECTIONS,
      landmark_drawing_spec=drawing_spec,
      connection_drawing_spec=drawing_spec)
  mp_drawing.draw_landmarks(
      image=annotated_image, 
      landmark_list=results.pose_landmarks, 
      connections=mp_holistic.POSE_CONNECTIONS,
      landmark_drawing_spec=drawing_spec,
      connection_drawing_spec=drawing_spec)
  cv2.imwrite(name, annotated_image)

import cv2

from google.colab.patches import cv2_imshow

import numpy as np

import glob

# image file names to files in list format

files=[]

for name in sorted(glob.glob('./images/*.png')):

files.append(name)

# Read images with OpenCV.

images = {name: cv2.imread(name) for name in files}

for name, image in images.items():

# Convert the BGR image to RGB and process it with MediaPipe Pose.

results = holistic.process(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))

# Draw pose landmarks.

annotated_image = image.copy()

mp_drawing.draw_landmarks(annotated_image, results.left_hand_landmarks, mp_holistic.HAND_CONNECTIONS)

mp_drawing.draw_landmarks(annotated_image, results.right_hand_landmarks, mp_holistic.HAND_CONNECTIONS)

mp_drawing.draw_landmarks(

image=annotated_image,

landmark_list=results.face_landmarks,

connections=mp_holistic.FACE_CONNECTIONS,

landmark_drawing_spec=drawing_spec,

connection_drawing_spec=drawing_spec)

mp_drawing.draw_landmarks(

image=annotated_image,

landmark_list=results.pose_landmarks,

connections=mp_holistic.POSE_CONNECTIONS,

landmark_drawing_spec=drawing_spec,

connection_drawing_spec=drawing_spec)

cv2.imwrite(name, annotated_image)

　最後に、images フォルダーの中にある検出結果を読み込んでmp4動画を作成します。

# 既に output.mp4 があれば削除
import os
if os.path.exists('./output.mp4'):
   os.remove('./output.mp4')

! ffmpeg -r 10 -i images/%6d.png\
               -vcodec libx264 -pix_fmt yuv420p output.mp4

# 既に output.mp4 があれば削除

import os

if os.path.exists('./output.mp4'):

os.remove('./output.mp4')

! ffmpeg -r 10 -i images/%6d.png\

-vcodec libx264 -pix_fmt yuv420p output.mp4

# --- mp4動画の再生 ---
from IPython.display import HTML
from base64 import b64encode

mp4 = open('./output.mp4', 'rb').read()
data_url = 'data:video/mp4;base64,' + b64encode(mp4).decode()
HTML(f"""
<video width="85%" height="85%" controls>
      <source src="{data_url}" type="video/mp4">
</video>""")

# --- mp4動画の再生 ---

from IPython.display import HTML

from base64 import b64encode

mp4 = open('./output.mp4', 'rb').read()

data_url = 'data:video/mp4;base64,' + b64encode(mp4).decode()

HTML(f"""

</video>""")

　顔と手は、かなり細かい検出がされていることが分かります。写っている範囲が狭いので、ポーズは肩の線のみ見えています。作成したmp4動画は、左側のファイル欄にある ./mediapipe/output.mp4 を右クリックするとダウンロードできます。

　ビデオ指定のところを、video_path = ‘./video/mana.mp4’ に変更して、それ以降のコードを実行すると、

　上半身が写ると、ポーズも検出していることが分かります。シングルパーソン用のため、より大きく写っている人物を優先して検出するので、後ろの男性は検出されません。

　ビデオ指定のところを、video_path = ‘./video/dance.mp4’ に変更して、それ以降のコードを実行すると、

　全身を写した場合の検出例です。一時的に右手の検出が外れますが、まずまずではないでしょうか。

4.コード（PCのカメラから検出）

※ご注意

　以下のコードは、google colabでは動作しません。ご自分のPCに、このリンクを参考にJupyter Notebookをインストールして実行して下さい。また、Git(コードをコピーするライブラリー)をインストールしてなければ、このリンクを参考に合わせてインストールして下さい。

　いよいよ、メインのリアルタイム検出です。カメラ付きのPCで Jupyter Nootbook を開いて、適当なフォルダーに移動し、下記のコードを実行します。GPUは不要です。

# MediaPipeをインストール
!pip install mediapipe

# Gitgub からコードをコピー
!git clone https://github.com/cedro3/mediapipe.git
%cd mediapipe/

# MediaPipeをインストール

!pip install mediapipe

# Gitgub からコードをコピー

!git clone https://github.com/cedro3/mediapipe.git

%cd mediapipe/

　ホリスティック検出（顔・手・ポーズの統合検出）をする場合は、下記のコードを実行するだけです。左上にFPSが表示されています。私のPCは3年前に買った MacbookAir ですが、それでも10FPSくらい出ます。凄い！

!python sample_holistic.py

1	!python sample_holistic.py

　顔の検出をする場合は、下記のコードを実行します。顔だけだと、20〜30FPSくらいでしょうか。

!python sample_face.py

1	!python sample_face.py

　手の検出をする場合は、下記のコードを実行します。手だけだと、15FPSくらいでしょうか。

!python sample_hand.py

1	!python sample_hand.py

　ポーズの検出をする場合は、下記のコードを実行します。ポーズだけだと、15FPSくらいでしょうか。

!python sample_pose.py

1	!python sample_pose.py

　MediaPipeの狙いはエッジデバイスでの検出です。特に、スマホで、顔・手・ポーズなどの検出を使って色々な応用が出来そうですね。

　では、また。

（Googleオリジナル） https://mediapipe.dev/index.html

（Githubオリジナル）https://github.com/Kazuhito00/mediapipe-python-sample

（Twitter投稿）

ディープラーニングで、動画から顔・手・ポーズの検出をやってみる

ブログ：https://t.co/R8mcOvL3zf pic.twitter.com/bdqH65B0pM
— cedro (@jun40vn) April 7, 2021

13 件のコメント

%cd mediapipe/ の後、!python sample_holistic.py
を実行したが、何も表示されません。他の3つも同じです。webカメラのライトが正常なら2つ点灯するのですが、最初から点灯している1つしか点灯していません。

返信する

cedro より:

2021年1月13日 8:49 PM

bravoastroさん
コメントありがとうございます。
詳しい状況を知りたいので、下記3点教えて下さい。

１）下記を実行した時に、返って来るコメントを教えて下さい。
————
# MediaPipeをインストール
!pip install mediapipe
# Gitgub からコードをコピー
!git clone https://github.com/cedro3/mediapipe.git
%cd mediapipe/
————

２）下記を実行した時に、返って来るコメントを教えて下さい。
——-
!python sample_holistic.py
——-

３）ご自分で設定したフォルダーの中に、mediapipeというファルダーは出来ていますでしょうか。

返信する
- bravoastro より:
  
  2021年1月20日 11:03 PM
  
  !python sample_holistic.py を実行した時のメッセージを次に示します。
  [ WARN:0] global /io/opencv/modules/videoio/src/cap_v4l.cpp (802) open VIDEOIO ERROR: V4L: can’t open camera by index 0
  INFO: Created TensorFlow Lite XNNPACK delegate for CPU.
  
  また、「自分で設定したフォルダー」と言う意味が分かりません。%cd mediapipe を実行しても何のエラーも出ないので出来ているのではないでしょうか？
  
  返信する
  - cedro より:
    
    2021年1月21日 8:57 AM
    
    bravoastro さん
    
    ご連絡ありがとうございます。
    !python sample_holistic.py を実行した時のメッセージを拝見しますと、google colabで動かそうとされている様です。このコードは、google colabでは動作しません。
    
    ブログにあります様に、リンク(https://qiita.com/tttamaki/items/0d4fc01c10dd40a13552)をご覧頂き、PCにJupyter Notebook をインストールして実行して下さい。
    また、Git(コードをコピーするライブラリー)をインストールしてなければ、このリンク（https://qiita.com/libra219/items/e49f9cc833ad9bb5d1ca)を参考に合わせてインストールして下さい。
    
    ＊「自分で設定したフォルダー」というは、!git clone する時のディレクトリという意味です。分かり難くてすみません。
    
    返信する

bravoastro さん

ブログの該当部分が勘違いし易いように思いましたので、修正します。
ありがとうございます。

返信する

cedro様

姿勢推定について最近調べ始めたAIも全く分からない素人です。
今週になってCOLAB等で姿勢推定が比較的簡単にできることを知りました。

さて、拝見したコードをそのままコピペして試していますが、MP4を自分で用意したものに変えて動かしてみたいのですが、ファイルをPCのどこに保存すればよいのか分かりません。

# ビデオの指定
video_path = ‘./video/satomi.mp4’

この部分を変更すればよいと記載があるものの、実際にファイルをPCのどこに格納すればよいかが分かりません。

ご教示賜れますでしょうか。
よろしくお願いいたします。

返信する

諸橋裕平 より:

2021年6月28日 9:39 PM

! python sample_holistic.py
を使おうと思ったとき
Traceback (most recent call last):
File “sample_holistic.py”, line 662, in
main()
File “sample_holistic.py”, line 58, in main
holistic = mp_holistic.Holistic(
TypeError: __init__() got an unexpected keyword argument ‘upper_body_only’
[ WARN:0] global C:\Users\runneradmin\AppData\Local\Temp\pip-req-build-k1ohfcms\opencv\modules\videoio\src\cap_msmf.cpp (438) `anonymous-namespace’::SourceReaderCB::~SourceReaderCB terminating async callback
というエラーがおきました。
upper_body_onlyが必要ないhand.pyのみうまくいきましたが、それ以外はこのエラーが起きてしまいます。
理由などが分かれば返信よろしくお願いいたします。

返信する
- cedro より:
  
  2021年6月29日 10:25 AM
  
  諸橋さん
  コメントありがとうございます。
  先ほど、こちらでも試してみましたがご連絡頂いたエラーが再現しません。
  ライブラリーの何かのバージョンが古いせいかもしれませんね。
  試しに、argparse のバージョンをアップデートしてみて下さい。
  
  返信する
cedro より:

2021年5月27日 10:08 AM

G3さん

コメントありがとうございます。
ブログにあるcolabリンクをお使いですよね。
その場合は、ご自分で用意されたmp4動画を、PCからドラッグ＆ドロップで、./mediapipe/video フォルダーにアップロードして下さい。あとは、video_path = ‘./video/＊＊＊.mp4 の＊＊＊.mp4 をご自分で用意されたファイル名に変更すればOKです。

返信する
- g3 より:
  
  2021年5月28日 7:00 AM
  
  cedro様
  
  お世話になります。
  家でやってみた所、うまくできました。
  ありがとうございました。
  
  返信する
  - cedro より:
    
    2021年5月28日 8:11 AM
    
    G3さん
    
    それは良かったです。^^
    
    返信する

こんにちはこんばんは、このエラーが発生します。
私はすでにargparseをpycharm1.4を含む最新バージョンに更新しました、そしてそれは出続けます、私は何ができますか？

ファイル “C：\ Users \ JUnty \ Desktop \ Tska \ RULA \ Holistic.py”、58行目、メイン
holistic = mp_holistic.Holistic（
TypeError：__ init __（）が予期しないキーワード引数 ‘upper_body_only’を取得しました

返信する