denseposeでU.S.A.

youtu.be

完全に出遅れたけど、DensePose使ってみたのでメモ。 Facebook等が発表した、2次元画像から人体の3Dサーフェスモデルを推定しちゃうすごい手法。

詳細とか使い方は下記のページあたりを参照。

shiropen.com github.com

環境

• ubuntu 16.04
• nvidia-docker導入済み（nvidia-dockerインストール、コンテナ等メモ - whoopsidaisies's diary）

導入

偉い人がdockerhubでイメージ公開してくれていたので活用。 動画ファイルをffmpegとかで画像にばらしておいた。

# ubuntu 16.04の端末で
nvidia-docker run -it -d --name densepose garyfeng/densepose
docker exec -it densepose bash
# コンテナ内
#   実行時の引数でフォルダを指定するとフォルダ内の画像を一気に処理してくれる
tools/infer_simple.py \
   --cfg configs/DensePose_ResNet101_FPN_s1x-e2e.yaml \
   --output-dir /root/data/infer_out/  \
   --image-ext png \
   --wts https://s3.amazonaws.com/densepose/DensePose_ResNet101_FPN_s1x-e2e.pkl \
   /root/data/frame

以上、簡単にできてすごい。