質問編集履歴
1
コード内容
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
|
-
|
|
49
|
+
while True:
|
|
50
|
-
|
|
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
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
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):
|