Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/kazuhito00/movenet-python-example
MoveNetを用いたPythonでの姿勢推定のデモ
https://github.com/kazuhito00/movenet-python-example
movenet movenet-lightning movenet-thunder multipose onnx singlepose tensorflow tensorflow-lite
Last synced: 3 months ago
JSON representation
MoveNetを用いたPythonでの姿勢推定のデモ
- Host: GitHub
- URL: https://github.com/kazuhito00/movenet-python-example
- Owner: Kazuhito00
- License: apache-2.0
- Created: 2021-08-23T15:54:44.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2022-09-23T05:38:37.000Z (over 2 years ago)
- Last Synced: 2023-03-04T13:14:50.037Z (almost 2 years ago)
- Topics: movenet, movenet-lightning, movenet-thunder, multipose, onnx, singlepose, tensorflow, tensorflow-lite
- Language: Python
- Homepage:
- Size: 68.8 MB
- Stars: 43
- Watchers: 3
- Forks: 8
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# MoveNet-Python-Example
[MoveNet](https://tfhub.dev/s?q=MoveNet)のPythonでの動作サンプルです。
ONNXに変換したモデルも同梱しています。変換自体を試したい方は[MoveNet_tf2onnx.ipynb](MoveNet_tf2onnx.ipynb)を使用ください。
2021/08/24時点でTensorFlow Hubで提供されている以下モデルを使用しています。
* [movenet/singlepose/lightning(v4)](https://tfhub.dev/google/movenet/singlepose/lightning/4)
* [movenet/singlepose/thunder(v4)](https://tfhub.dev/google/movenet/singlepose/thunder/4)
* [movenet/multipose/lightning(v1)](https://tfhub.dev/google/movenet/multipose/lightning/1)
# Requirement
* TensorFlow 2.3.0 or later
* tensorflow-hub 0.12.0 or later
* OpenCV 3.4.2 or later
* onnxruntime 1.5.2 or later ※ONNX推論を使用する場合のみ
# Demo
デモの実行方法は以下です。
#### SignlePose
```bash
python demo_singlepose.py
```
* --device
カメラデバイス番号の指定
デフォルト：0
* --file
動画ファイルの指定 ※指定時はカメラデバイスより優先
デフォルト：指定なし
* --width
カメラキャプチャ時の横幅
デフォルト：960
* --height
カメラキャプチャ時の縦幅
デフォルト：540
* --mirror
VideoCapture()取り込みデータを左右反転するか否か
デフォルト：指定なし
* --model_select
使用モデルの選択
Saved Model, ONNX：0→Lightning 1→Thunder
TFLite：0→Lightning(float16) 1→Thunder(float16) 2→Lightning(int8) 3→Thunder(int8)
デフォルト：0
* --keypoint_score
キーポイント表示の閾値
デフォルト：0.4
#### MultiPose
```bash
python demo_multipose.py
```
* --device
カメラデバイス番号の指定
デフォルト：0
* --file
動画ファイルの指定 ※指定時はカメラデバイスより優先
デフォルト：指定なし
* --width
カメラキャプチャ時の横幅
デフォルト：960
* --height
カメラキャプチャ時の縦幅
デフォルト：540
* --mirror
VideoCapture()取り込みデータを左右反転するか否か
デフォルト：指定なし
* --keypoint_score
キーポイント表示の閾値
デフォルト：0.4
* --bbox_score
バウンディングボックス表示の閾値
デフォルト：0.2
# Reference
* [TensorFlow Hub：MoveNet](https://tfhub.dev/s?q=MoveNet)
# Author
高橋かずひと(https://twitter.com/KzhtTkhs)
# License
MoveNet-Python-Example is under [Apache-2.0 License](LICENSE).
# License(Movie)
サンプル動画は[NHKクリエイティブ・ライブラリー](https://www.nhk.or.jp/archives/creative/)の[ストリートバスケット](https://www2.nhk.or.jp/archives/creative/material/view.cgi?m=D0002080169_00000)を使用しています。