tf-pose-estimationをGoogleColaboratoryでやる

準備

tf-pose-estimation/run.py at master · ildoonet/tf-pose-estimation · GitHub
をダウンロードする。ファイル名はrun.pyにする。
run.pyをエディタで開いて88行目付近を

plt.colorbar()
plt.savefig('output.png')
plt.show()

に編集して保存する。

実行

!git clone https://www.github.com/ildoonet/tf-pose-estimation
cd tf-pose-estimation
pip install tqdm==4.29.0
!pip install -r requirements.txt
!bash models/graph/cmu/download.sh
!sudo apt install swig
!swig -python -c++ tf_pose/pafprocess/pafprocess.i
cd tf_pose/pafprocess
!python setup.py build_ext --inplace
cd ../../
rm run.py
from google.colab import files
files.upload()
!python run.py --model=cmu --resize=432x368 --image=./images/p1.jpg
from IPython.display import Image,display_png
display_png(Image('output.png'))

uploadのところで準備で編集したrun.pyをアップロードする。