質問するログイン新規登録

質問編集履歴

1

コード内容

2022/10/12 03:21

投稿

ke0313
ke0313

スコア65

title CHANGED
File without changes
body CHANGED
@@ -46,18 +46,23 @@
46
46
 
47
47
  def getSpecificData(filelist):
48
48
  for i in range(len(filelist)):
49
- with open(filelist[i]) as f:
49
+ while True:
50
- data = json.load(f)
50
+ try:
51
- data = np.array(data['people'][0]['pose_keypoints_2d']).reshape(-1,3)
52
- df = pd.DataFrame(data, columns=['X','Y','P'], index=["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \
53
- "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"])
54
51
 
55
- # 自分の必要なデータを取り出す
56
- writeCSV([float(df.at["Nose", "X"]),float(df.at["Nose", "Y"]),float(df.at["Neck", "X"]),float(df.at["Neck", "Y"]),float(df.at["RShoulder", "X"]),float(df.at["RShoulder", "Y"]),float(df.at["RElbow", "X"]),float(df.at["RElbow", "Y"]), float(df.at["RWrist", "X"]), float(df.at["RWrist", "Y"]) ,float(df.at["LShoulder", "X"]),float(df.at["LShoulder", "Y"]),float(df.at["LElbow", "X"]) \
57
- , float(df.at["LElbow", "Y"]), float(df.at["LWrist", "X"]), float(df.at["LWrist", "Y"]), float(df.at["MidHip", "X"]), float(df.at["MidHip", "Y"]), float(df.at["LWrist", "X"]), float(df.at["LWrist", "Y"]), float(df.at["RHip", "X"]), float(df.at["RHip", "Y"]), float(df.at["RKnee", "X"]), float(df.at["RKnee", "Y"]), float(df.at["RAnkle", "X"]), float(df.at["RAnkle", "Y"]), float(df.at["LHip", "X"]), float(df.at["LHip", "Y"])\
58
- , float(df.at["LKnee", "X"]), float(df.at["LKnee", "Y"]), float(df.at["LAnkle", "X"]), float(df.at["LAnkle", "Y"]), float(df.at["REye", "X"]), float(df.at["REye", "Y"]), float(df.at["LEye", "X"]), float(df.at["LEye", "Y"]), float(df.at["REar", "X"]), float(df.at["REar", "Y"]), float(df.at["LEar", "X"]), float(df.at["LEar", "Y"]), float(df.at["LBigToe", "X"]), float(df.at["LBigToe", "Y"])\
59
- , float(df.at["LSmallToe", "X"]), float(df.at["LSmallToe", "Y"]), float(df.at["LHeel", "X"]), float(df.at["LHeel", "Y"]), float(df.at["RBigToe", "X"]), float(df.at["RBigToe", "Y"]), float(df.at["RSmallToe", "X"]), float(df.at["RSmallToe", "Y"]), float(df.at["RHeel", "X"]), float(df.at["RHeel", "Y"])])
52
+ with open(filelist[i]) as f:
53
+ data = json.load(f)
54
+ data = np.array(data['people'][0]['pose_keypoints_2d']).reshape(-1,3)
55
+ df = pd.DataFrame(data, columns=['X','Y','P'], index=["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \
56
+ "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"])
60
57
 
58
+ # 自分の必要なデータを取り出す
59
+ writeCSV([float(df.at["Nose", "X"]),float(df.at["Nose", "Y"]),float(df.at["Neck", "X"]),float(df.at["Neck", "Y"]),float(df.at["RShoulder", "X"]),float(df.at["RShoulder", "Y"]),float(df.at["RElbow", "X"]),float(df.at["RElbow", "Y"]), float(df.at["RWrist", "X"]), float(df.at["RWrist", "Y"]) ,float(df.at["LShoulder", "X"]),float(df.at["LShoulder", "Y"]),float(df.at["LElbow", "X"]) \
60
+ , float(df.at["LElbow", "Y"]), float(df.at["LWrist", "X"]), float(df.at["LWrist", "Y"]), float(df.at["MidHip", "X"]), float(df.at["MidHip", "Y"]), float(df.at["LWrist", "X"]), float(df.at["LWrist", "Y"]), float(df.at["RHip", "X"]), float(df.at["RHip", "Y"]), float(df.at["RKnee", "X"]), float(df.at["RKnee", "Y"]), float(df.at["RAnkle", "X"]), float(df.at["RAnkle", "Y"]), float(df.at["LHip", "X"]), float(df.at["LHip", "Y"])\
61
+ , float(df.at["LKnee", "X"]), float(df.at["LKnee", "Y"]), float(df.at["LAnkle", "X"]), float(df.at["LAnkle", "Y"]), float(df.at["REye", "X"]), float(df.at["REye", "Y"]), float(df.at["LEye", "X"]), float(df.at["LEye", "Y"]), float(df.at["REar", "X"]), float(df.at["REar", "Y"]), float(df.at["LEar", "X"]), float(df.at["LEar", "Y"]), float(df.at["LBigToe", "X"]), float(df.at["LBigToe", "Y"])\
62
+ , float(df.at["LSmallToe", "X"]), float(df.at["LSmallToe", "Y"]), float(df.at["LHeel", "X"]), float(df.at["LHeel", "Y"]), float(df.at["RBigToe", "X"]), float(df.at["RBigToe", "Y"]), float(df.at["RSmallToe", "X"]), float(df.at["RSmallToe", "Y"]), float(df.at["RHeel", "X"]), float(df.at["RHeel", "Y"])])
63
+
64
+ except IndexError:
65
+ pass
61
66
 
62
67
 
63
68
  def writeCSV(data):