python
import json import pandas as pd import numpy as np import glob import csv def getFileName(path): filelist = glob.glob(path+ "/*.json") return filelist def getSpecificData(filelist): for i in range(len(filelist)): while True: try: with open(filelist[i]) as f: data = json.load(f) data = np.array(data['people'][0]['pose_keypoints_2d']).reshape(-1,3) df = pd.DataFrame(data, columns=['X','Y','P'], index=["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \ "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"]) # 自分の必要なデータを取り出す 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"]) \ , 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"])\ , 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"])\ , 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"])]) except IndexError: pass def writeCSV(data): with open('openposetameshi.csv', 'a') as f: #変更 writer = csv.writer(f, lineterminator='\n') writer.writerow(data) def main(): filelist = getFileName(input("JSONのディレクトリのパスを入力: ")) with open('openposetameshi.csv', 'w') as f: #変更 writer = csv.writer(f, lineterminator='\n') # 自分の必要なデータの列の名前を用意。上のデータと同じだけの列数を揃える。 writer.writerow(["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \ "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"]) getSpecificData(filelist) if __name__ == '__main__': main()
0 コメント