OpenPoseの導入:ModuleNotFoundError: No module named 'tf_pose' の解決

Question

前提・実現したいこと

GoogleColaboratoryでOpenPoseを導入しようとしています。

発生している問題・エラーメッセージ

あるソースコードを実行する際にモジュールのインポートエラーが発生しました。

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-25-398741285e53> in <module>()
      5 #import tf_pose
      6 
----> 7 from tf_pose import common
      8 import cv2
      9 import numpy as np

ModuleNotFoundError: No module named 'tf_pose'

該当のソースコード

import argparse
import sys
import time
import math
#import tf_pose

from tf_pose import common
import cv2
import numpy as np
from tf_pose.estimator import TfPoseEstimator
from tf_pose.networks import get_graph_path, model_wh

import glob
import inspect

images_path=image01
path_list=glob.glob(images_path+"/*")

#画像で最も大きい人間のボーンを取得する関数
def getBiggestHuman(humans):
    maxNum=0
    maxHuman=humans[0]

    for human in humans:
        tempNum=0

        #パーツインデックス取得
        keys=list(human.body_parts.keys())
        temp=human.body_parts[keys[0]]
        tempX=temp.x
        tempY=temp.y
        keys = keys[1:]
        for i in keys:
            parts = human.body_parts[i]
            tempNum += math.sqrt((tempX-parts.x)**2 + (tempY-parts.y)**2 )

        if tempNum > maxNum:
            maxNum=tempNum
            maxHuman=human
    return maxHuman

#pix2pix用の画像を作成する関数
def makeData(humans, orig_image, save_path):
    hu = getBiggestHuman(humans)
    kuronuri_image=np.zeros_like(orig_image)
    if len(hu.body_parts.keys()) >= 17:
        image = TfPoseEstimator.draw_humans(kuronuri_image, [hu], imgcopy=False)
        image=cv2.hconcat([orig_image,image])
        cv2.imwrite(save_path,image)

w, h = 256,256#ここは1で作った画像の縦横サイズを指定

e = TfPoseEstimator(get_graph_path('cmu'), target_size=(w, h))


for path in path_list:
    orig_image = common.read_imgfile(path, None, None)
    humans = e.inference(orig_image, resize_to_default=(w > 0 and h > 0), upsample_size=4.0)
    save_path =train
    makeData(humans=humans, orig_image=orig_image, save_path=save_path)

試したこと

openposeは以下の手順でインストールしました。

!git clone https://github.com/ildoonet/tf-pose-estimation

import os 
os.chdir('tf-pose-estimation')
!ls
!pwd

!pip install -r requirements.txt

!apt-get -q -y install swig 

os.chdir('tf_pose/pafprocess')
!ls
!pwd

!swig -python -c++ pafprocess.i && python setup.py build_ext --inplace

!git clone https://www.github.com/ildoonet/tf-pose-estimation

os.chdir('tf-pose-estimation')

!python setup.py install

モジュールを正しい場所へ移動できていないことが原因と考えています。もしそうならば、正しい場所へどのように移したらよいのか分かりません。

補足情報（FW/ツールのバージョンなど）

以下のGitHubを参考に進めました。
tf-pose-estimation

Answer 1

特にimportエラーは発生しませんでした。最後に表示されている Tensorflow のメッセージは警告で、「もうすぐコラボのデフォルトのTensorflowバージョンをv2.0に切り替えるよ。」と表示されているだけですので無視してください。今回の問題は、最後にプログラムを実行する階層が、/content/tf-pose-estimation ではなかったために、tf_pose フォルダが探せなかっただけだと思います。下記に手順を見直したサンプルのノートブックを記載しておきます。あと、最後の !git clone https://www.github.com/ildoonet/tf-pose-estimation は適切にカレントパスが変更されていれば実行不要です。
OpenPose Sample Notebook

Comments

[Mattcha]

無事に実行できました！
ありがとうございます！