tfpose-estimationというもので骨格推定を行おうとすると下記のようなエラーがでてしまいます。解決方法を教えていただきたいです。
[エラー]
Traceback (most recent call last):
File "run1.py", line 49, in <module>
fname = os.path.basename(path_list)
File "/usr/lib/python3.6/posixpath.py", line 146, in basename
p = os.fspath(p)
TypeError: expected str, bytes or os.PathLike object, not list
問題となっているrun1.py下記のようなプログラムです。
path_list = glob.glob('test/No_10000/No_10000_a_00000029_.png')
で画像を読み込み
cv2.imwrite('./result/No_10000_a_00000029_.png', image)
で./result以下に骨格推定結果を保存するというようなプログラムです
run1.py
1import argparse 2import logging 3import sys 4import time 5import glob 6import os 7 8from tf_pose import common 9import cv2 10import numpy as np 11from tf_pose.estimator import TfPoseEstimator 12from tf_pose.networks import get_graph_path, model_wh 13 14logger = logging.getLogger('TfPoseEstimatorRun') 15logger.handlers.clear() 16logger.setLevel(logging.DEBUG) 17ch = logging.StreamHandler() 18ch.setLevel(logging.DEBUG) 19formatter = logging.Formatter('[%(asctime)s] [%(name)s] [%(levelname)s] %(message)s') 20ch.setFormatter(formatter) 21logger.addHandler(ch) 22 23path_list = glob.glob('test/No_10000/No_10000_a_00000029_*.png') 24 25 26if __name__ == '__main__': 27 parser = argparse.ArgumentParser(description='tf-pose-estimation run') 28 parser.add_argument('--image', type=str, default='./images/p1.png') 29 parser.add_argument('--image_dir', type=str,default=glob.glob('./images/*.png')) 30 parser.add_argument('--model', type=str, default='cmu', 31 help='cmu / mobilenet_thin / mobilenet_v2_large / mobilenet_v2_small') 32 parser.add_argument('--resize', type=str, default='0x0', 33 help='if provided, resize images before they are processed. ' 34 'default=0x0, Recommends : 432x368 or 656x368 or 1312x736 ') 35 parser.add_argument('--number_people_max', type=int, default=1, help='maximum number of people') 36 parser.add_argument('--resize-out-ratio', type=float, default=4.0, 37 help='if provided, resize heatmaps before they are post-processed. default=1.0') 38 parser.add_argument('--output_json', type=str, default='/tmp/', help='writing output json dir') 39 args = parser.parse_args() 40 41 w, h = model_wh(args.resize) 42 if w == 0 or h == 0: 43 e = TfPoseEstimator(get_graph_path(args.model), target_size=(432, 368)) 44 else: 45 e = TfPoseEstimator(get_graph_path(args.model), target_size=(w, h)) 46 47 # estimate human poses from a single image ! 48 image = cv2.imread(args.image, None) 49 fname = os.path.basename(path_list) 50 if image is None: 51 logger.error('Image can not be read, path=%s' % args.image) 52 sys.exit(-1) 53 t = time.time() 54 humans = e.inference(image, resize_to_default=(w > 0 and h > 0),upsample_size=args.resize_out_ratio) 55 elapsed = time.time() - t 56 57 logger.info('inference image: %s in %.4f seconds.' % (args.image, elapsed)) 58 59 image = TfPoseEstimator.draw_humans(image, humans, imgcopy=False, frame=0, output_json_dir=args.output_json) 60 cv2.imwrite('./result/No_10000_a_00000029_*.png', image)
あなたの回答
tips
プレビュー