一、把数据存入 JSON文件
# Data is stored in json format
import os
import json
def json_file():
data_info = []
frame_data = {"frame_index": 25}
skeleton = {}
corrdinates = [0.5, 0.5]
socre = [0.86]
skeleton["pose"] = corrdinates
skeleton["score"] = socre
skeletons = [skeleton]
frame_data["skeleton"] = skeletons
data_info = [frame_data]
data = dict()
data["data"] = data_info
data["label"] = "running"
data["label_index"] = 1
return data
json_file_path = "./save_data.json"
save_data = json_file()
if not os.path.exists(json_file_path):
os.makedirs(json_file_path)
with open(json_file_path, 'w') as outfile:
json.dump(save_data, outfile)
json文件中的内容为:
二、从JSON文件读取数据
# Read data in json file
import json
def read_json(filename):
info = json.load(open(filename))
data_info = info["data"]
frame_index = data_info["frame_index"]
skeletons = data_info["skeleton"]
coordinates, socre = [], []
coordinates = skeletons["pose"]
socre = skeletons["score"]
label = info["label"]
label_index = info["label_index"]
print("frame_index: ", frame_index)
print("coordinates: ", coordinates)
print("socre: ", socre)
print("label: ", label)
print("label_index: ", label_index)
json_file_path = "./save_data.json"
read_json(json_file_path)
输出结果: