pyファイルが上手くインポートできない。

Question

pyファイルをインポートして、実行したい。

openpose
上記URLのpose_detector.pyを以下のプログラムでインポートしました。

import sys
# モジュール検索パスの設定
REPO_ROOT = '/Users/Chainer_Realtime_Multi-Person_Pose_Estimation'
sys.path.append(REPO_ROOT)

# PoseDetectorクラスのインポート
from pose_detector import PoseDetector

# モデルのロード
arch_name = 'posenet'
image_path = os.path.join(REPO_ROOT, 'data', 'person.png')
weight_path = os.path.join(REPO_ROOT, 'models', 'coco_posenet.npz')

model = PoseDetector(arch_name, weight_path)

---------------------------------------------------------------------------

Loading the model...

と表示された後に以下を実行すると

poses, _ = pose_detector(img)
img = draw_person_pose(img, pose_arr)

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-53-d818c191d81a> in <module>()
----> 1 poses, _ = pose_detector(img)
      2 img = draw_person_pose(img, pose_arr)

NameError: name 'pose_detector' is not defined

となって、うまく呼び出せません。
ご回答よろしくおねがいします・

Answer 1

pose_detector.pyのif __name__ == '__main__':以下の記述を見るに、

model = PoseDetector(arch_name, weight_path)

としたのなら

poses, _ = model(img)

と実行する必要があるのでは？

そして次はdraw_person_poseが見つからないというエラーが出る気がするので、

from pose_detector import draw_person_pose

も追加しておいてください。

Comments

[sasakikik]

できました！
ありがとうございます。