Tensorflow.js 学習済みモデルを使って人物検出(人数)を行う
TensorFlow.js
の学習済みモデルを使って人物検出を行い、画像に人物が何人いるかを判定する方法を紹介します。
今回はBodyPix
というライブラリ(学習済みモデル)を使用します。
BodyPix
は、人数判定だけでなく、検出した人物にフィルターをかけたり、人物の頭、腕など、細かく分類したり、とても高機能なライブラリです。
https://github.com/tensorflow/tfjs-models/tree/master/body-pix
パッケージのインストール
以下のコマンドで必要なパッケージをインストールします。
$ npm install @tensorflow-models/body-pix
予測
以下のように、予測を行います。
const bodyPix = require('@tensorflow-models/body-pix');
async function predict() {
// モデルの読み込み
const model = await bodyPix.load();
// 判定する画像要素を取得
const img = document.getElementById('img');
// 予測
const segmentation = await model.segmentMultiPerson(img);
// 人数を表示
console.log(segmentation.length);
}
segmentMultiPerson
メソッドの引数には、画像(img
要素)だけでなく、キャンバス(canvas
要素)を指定することも可能です。
取得結果segmentation
には検出した座標などの様々な情報を人物ごとに配列で格納されています。
今回は単純に人数を取得するため、配列の長さを取得しています。