KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#developer-tools #typescript #dsh #dsh-plugin #ai-agent #codex #ai-agents #deepseek-harness #claude-code #cli #react #mcp

mediapipe-python-sample

★ 339 Open GitHub ↗

MediaPipeのPythonパッケージのサンプルです。2024/9/1時点でPython実装のある15機能について用意しています。

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

> [!IMPORTANT]
> MediaPipe レガシーソリューションのサポートは、2023年3月1日で終了しています。 
> 従来のソリューションのサンプルは [_legacy](_legacy)ディレクトリに移動しました。 
> MediaPipeは後方互換を保っており、現パッケージでもレガシーソリューションのサンプルを実行出来ます。

# mediapipe-python-sample
[google-ai-edge/mediapipe](https://github.com/google-ai-edge/mediapipe)のPythonパッケージのサンプルスクリプト集です。 
2024/9/1時点でPython実装のある以下15機能について用意しています。
* [物体検出（Object Detection）](https://ai.google.dev/edge/mediapipe/solutions/vision/object_detector?hl=ja)
* [画像分類（Image Classification）](https://ai.google.dev/mediapipe/solutions/vision/image_classifier?hl=ja)
* [画像セグメンテーション（Image Segmentation）](https://ai.google.dev/mediapipe/solutions/vision/image_segmenter?hl=ja)
* [インタラクティブ セグメンテーション（Interactive segmentation）](https://ai.google.dev/mediapipe/solutions/vision/interactive_segmenter?hl=ja)
* [手検出（Hand Landmark detection）](https://ai.google.dev/mediapipe/solutions/vision/hand_landmarker?hl=ja)
* [手のジェスチャー認識（Gesture Recognition）](https://ai.google.dev/mediapipe/solutions/vision/gesture_recognizer?hl=ja)
* [画像の埋め込み表現（Image Embedding）](https://ai.google.dev/mediapipe/solutions/vision/image_embedder?hl=ja)
* [顔検出（Face Detection）](https://ai.google.dev/mediapipe/solutions/vision/face_detector?hl=ja)
* [顔のランドマーク検出（Face Landmark Detection）](https://ai.google.dev/mediapipe/solutions/vision/face_landmarker?hl=ja)
* [顔のスタイル変換（Face Stylization）](https://ai.google.dev/mediapipe/solutions/vision/face_stylizer?hl=ja)
* [姿勢推定（Pose Landmark Detection）](https://ai.google.dev/mediapipe/solutions/vision/pose_landmarker?hl=ja)
* [テキスト分類（Text Classification）](https://ai.google.dev/mediapipe/solutions/text/text_classifier?hl=ja)
* [テキストの埋め込み表現（Text Embedding）](https://ai.google.dev/mediapipe/solutions/text/text_embedder?hl=ja)
* [テキスト言語分類（Language Detector）](https://ai.google.dev/mediapipe/solutions/text/language_detector?hl=ja)
* [音分類（Audio Classification）](https://ai.google.dev/mediapipe/solutions/audio/audio_classifier?hl=ja)

# Requirement 
* mediapipe 0.10.14 or later
* opencv-python 4.10.0.84 or later
* tqdm 4.66.5 or later　※重みファイルダウンロードに使用
* requests 2.32.3 or later　※重みファイルダウンロードに使用
* scipy 1.14.1 or later　※音分類（Audio Classification）サンプルを実行する場合のみ
* numpy 1.26.4　※NumPyは1.x系

```
pip install -r requirements.txt
```

# Demo
デモの実行方法は以下です。

### 物体検出（Object Detection）
```bash
python sample_object_detection.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --model 
使用モデル[0, 1, 2, 3, 4, 5, 6, 7]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
[COCOデータセット](https://cocodataset.org/#home)でトレーニングされた重みで、サポートされているラベルは[labelmap.txt](https://storage.googleapis.com/mediapipe-tasks/object_detector/labelmap.txt) 
デフォルト：0 
 * 0:EfficientDet-Lite0(int8)
 * 1:EfficientDet-Lite0(float 16)
 * 2:EfficientDet-Lite0(float 32)
 * 3:EfficientDet-Lite2(int8)
 * 4:EfficientDet-Lite2(float 16)
 * 5:EfficientDet-Lite2float 32）
 * 6:SSDMobileNet-V2(int8)
 * 7:SSDMobileNet-V2(float 32)
* --score_threshold 
スコア閾値 
デフォルト：0.5
</details>
<img src="https://github.com/user-attachments/assets/049776c6-bf7c-47df-982b-8d2e21e1f322" loading="lazy" width="250px">

### 画像分類（Image Classification）
```bash
python sample_image_classification.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --model 
使用モデル[0, 1, 2, 3]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
[ImageNet](https://www.image-net.org/)でトレーニングされた重みで、サポートされているラベルは[labels.txt](https://storage.googleapis.com/mediapipe-tasks/image_classifier/labels.txt) 
デフォルト：0 
 * 0:EfficientNet-Lite0(int8)
 * 1:EfficientNet-Lite0(float 32)
 * 2:EfficientNet-Lite2(int8)
 * 3:EfficientNet-Lite2(float 32)
* --max_results 
結果出力数 
デフォルト：5
</details>
<img src="https://github.com/user-attachments/assets/78b2c7ef-83b9-47e3-9849-70d16bd7f241" loading="lazy" width="250px">

### 画像セグメンテーション（Image Segmentation）
```bash
python sample_image_segmentation.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --model 
使用モデル[0, 1, 2, 3, 4]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:SelfieSegmenter(square)
 * 1:SelfieSegmenter(landscape)
 * 2:HairSegmenter
 * 3:SelfieMulticlass(256x256)
 * 4:DeepLab-V3
</details>
<img src="https://github.com/user-attachments/assets/2033ba3b-3a66-437e-99d8-61fb38984b0b" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/93881e8c-cb19-4e14-b2f5-38bc061e481f" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/572f94d4-cbbc-4813-a408-f23f25abc1f9" loading="lazy" width="250px">

### インタラクティブ セグメンテーション（Interactive segmentation）
```bash
python sample_interactive_image_segmentation.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --image 
画像パスの指定 
デフォルト：asset/hedgehog01.jpg
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:MagicTouch
</details>
<img src="https://github.com/user-attachments/assets/04f987b2-67e0-49f1-ad78-c5086d1bb6d4" loading="lazy" width="250px">

### 手検出（Hand Landmark detection）
```bash
python sample_hand_landmarks_detection.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --unuse_mirror 
ミラー表示不使用 
デフォルト：指定なし
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:HandLandmarker (full)
* --num_hands 
検出数 
デフォルト：2
* --use_world_landmark 
ワールド座標表示 
デフォルト：指定なし
</details>
<img src="https://github.com/user-attachments/assets/0bc2d091-6eae-4029-9a4c-44ea151980fb" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/9946c203-93e9-4fac-9979-e1a752f67dc4" loading="lazy" width="250px">

### 手のジェスチャー認識（Gesture Recognition）
```bash
python sample_hand_gesture_recognition.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --unuse_mirror 
ミラー表示不使用 
デフォルト：指定なし
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
認識ジェスチャーは「Closed fist」「Open palm」「Pointing up」「Thumbs down」「Thumbs up」「Victory」「Love」「Unknown」 
デフォルト：0 
 * 0:HandGestureClassifier
</details>
<img src="https://github.com/user-attachments/assets/844dd2cf-3bf7-4803-8782-8fc2eb9bc3dc" loading="lazy" width="250px">

### 画像の埋め込み表現（Image Embedding）
```bash
python sample_image_embedding.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --image01 
画像パス1の指定 
デフォルト：asset/hedgehog01.jpg
* --image02 
画像パス2の指定 
デフォルト：asset/hedgehog02.jpg
* --model 
使用モデル[0, 1]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:MobileNet-V3 (small)
 * 1:MobileNet-V3 (large)
* --unuse_l2_normalize 
特徴ベクトルを L2 ノルムで正規化しない 
デフォルト：指定なし
* --unuse_quantize 
特徴ベクトルを スカラー量子化によってバイトに量子化しない 
デフォルト：指定なし
</details>
<img src="https://github.com/user-attachments/assets/58fea0f1-83fa-448c-a36e-e9c760669a61" loading="lazy" width="500px"> 
<img src="https://github.com/user-attachments/assets/d1d0718c-4bf7-4151-ab51-bdb4806a264e" loading="lazy" width="500px"> 
<img src="https://github.com/user-attachments/assets/c1e5a632-2000-4949-9914-feeafac2ae17" loading="lazy" width="500px"> 
<img src="https://github.com/user-attachments/assets/7e1e788b-dd01-4521-bb54-fbddd2b698e3" loading="lazy" width="500px">

### 顔検出（Face Detection）
```bash
python sample_face_detection.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:BlazeFace (short-range)
</details>
<img src="https://github.com/user-attachments/assets/10b79e31-7d0e-4d1e-9552-34b6ea9aa419" loading="lazy" width="250px">

### 顔のランドマーク検出（Face Landmark Detection）
```bash
python sample_face_landmark_detection.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:FaceLandscapeer
* --num_faces 
検出数 
デフォルト：1
* --unuse_output_face_blendshapes 
顔のブレンドシェイプを出力しない 
デフォルト：指定なし
* --unuse_output_facial_transformation_matrixes 
顔変換行列を出力しない 
デフォルト：指定なし
</details>
<img src="https://github.com/user-attachments/assets/e204f9f2-2284-4f5e-b26d-b846fc01e529" loading="lazy" width="250px">

### 顔のスタイル変換（Face Stylization）
```bash
python sample_face_stylization.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --model 
使用モデル[0, 1, 2]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:Color sketch
 * 1:Color ink
 * 2:Oil painting
</details>
<img src="https://github.com/user-attachments/assets/de975b72-7dab-4fe6-800e-99d68a42e939" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/3d224686-948a-48c2-bacb-52aa8285719e" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/9483b711-5c1c-40ca-8ae4-6e8f6beabb74" loading="lazy" width="250px">

### 姿勢推定（Pose Landmark Detection）
```bash
python sample_pose_landmark_detection.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --device 
カメラデバイス番号の指定 
デフォルト：0
* --video 
動画パスの指定 ※指定時はカメラより優先 
デフォルト：None
* --width 
カメラキャプチャ時の横幅 
デフォルト：960
* --height 
カメラキャプチャ時の縦幅 
デフォルト：540
* --unuse_mirror 
ミラー表示不使用 
デフォルト：指定なし
* --model 
使用モデル[0, 1, 2]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:Pose landmarker(lite)
 * 1:Pose landmarker(Full)
 * 2:Pose landmarker(Heavy)
* --use_output_segmentation_masks 
セグメンテーションを実施 
デフォルト：指定なし
* --use_world_landmark 
ワールド座標表示を実施 
デフォルト：指定なし
</details>
<img src="https://github.com/user-attachments/assets/54b752b9-8374-41d5-bd48-4cf76f803f18" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/9325e6e5-52df-4e8a-9e13-8c6d00d7ae37" loading="lazy" width="250px"> <img src="https://github.com/user-attachments/assets/ec001fa0-3042-4fb6-a653-633403e93691" loading="lazy" width="250px">

### テキスト分類（Text Classification）
```bash
python sample_text_classification.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --input_text 
入力テキスト 
デフォルト：I'm looking forward to what will come next.
* --model 
使用モデル[0, 1]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:BERT-classifier
 * 1:Average word embedding
</details>
<img src="https://github.com/user-attachments/assets/d40b6a51-b50a-4d5b-879f-0c5bfc1e62e5" loading="lazy" width="500px">

### テキストの埋め込み表現（Text Embedding）
```bash
python sample_text_embedding.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --input_text01 
入力テキスト1 
デフォルト：I'm feeling so good
* --input_text02 
入力テキスト2 
デフォルト：I'm okay I guess
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:Universal Sentence Encoder
* --unuse_l2_normalize 
特徴ベクトルを L2 ノルムで正規化しない 
デフォルト：指定なし
* --use_quantize 
特徴ベクトルを スカラー量子化によってバイトに量子化する 
デフォルト：指定なし
</details>
<img src="https://github.com/user-attachments/assets/6c82d941-98f3-4d91-8b9d-d2d7439de47e" loading="lazy" width="500px">

### テキスト言語分類（Language Detector）
```bash
python sample_text_language_detection.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --input_text 
入力テキスト 
デフォルト：分久必合合久必分
* --model 
使用モデル[0, 1]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:Language Detector
</details>
<img src="https://github.com/user-attachments/assets/537e341e-9534-4b4b-9204-d7c0ac31af21" loading="lazy" width="500px">

### 音分類（Audio Classification）
```bash
python sample_audio_classification.py
```
<details>
<summary>コマンドライン引数オプション</summary>
 
* --input_audio 
入力音声ファイルのパス 
デフォルト：asset/hyakuninisshu_02.wav
* --model 
使用モデル[0]　※対象モデルの重みがmodelディレクトリ内に無い場合ダウンロードを実行 
デフォルト：0 
 * 0:YamNet
* --max_results 
結果出力数 
デフォルト：5
</details>
<img src="https://github.com/user-attachments/assets/a0f42d85-15e5-490f-b305-c99d063251af" loading="lazy" width="500px">

# Reference
* [google-ai-edge/mediapipe](https://github.com/google-ai-edge/mediapipe)

# Author
高橋かずひと(https://twitter.com/KzhtTkhs)
 
# License 
mediapipe-python-sample is under [Apache-2.0 License](LICENSE).

# License(Image, Video, Audio)
サンプル実行用に格納している画像などは以下を利用しています。
* [ぱくたそ](https://www.pakutaso.com)様：[トゲトゲのサボテンとハリネズミ](https://www.pakutaso.com/20190257050post-19488.html)
* [ぱくたそ](https://www.pakutaso.com)様：[人間の靴にはまり込むハリネズ](https://www.pakutaso.com/20171041289post-13677.html)
* [ぱくたそ](https://www.pakutaso.com)様：[靴にすっぽり隠れるハリネズミ](https://www.pakutaso.com/20171039289post-13676.html)
* [NHKクリエイティブ・ライブラリー](https://www.nhk.or.jp/archives/creative/)様：「[猫カフェのネコ（３）](https://www2.nhk.or.jp/archives/movies/?id=D0002161325_00000)」
* [NHKクリエイティブ・ライブラリー](https://www.nhk.or.jp/archives/creative/)様：「[寅さんの像　アップ](https://www2.nhk.or.jp/archives/movies/?id=D0002022189_00000)」
* [NHKクリエイティブ・ライブラリー](https://www.nhk.or.jp/archives/creative/)様：「[音声 　百人一首　二](https://www2.nhk.or.jp/archives/movies/?id=D0002110102_00000)」

mediapipe-python-sample

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README