33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
import json
|
|
|
|
# 读取.py格式的配置文件并执行
|
|
def read_py_config(file_path):
|
|
# 创建一个空的字典用于保存配置
|
|
config_dict = {}
|
|
|
|
# 执行.py文件并将结果存入config_dict
|
|
with open(file_path, 'r') as f:
|
|
# 执行Python代码并保存配置
|
|
exec(f.read(), {}, config_dict)
|
|
|
|
return config_dict
|
|
|
|
# 将字典保存为json格式
|
|
def save_as_json(config_dict, output_file_path):
|
|
with open(output_file_path, 'w') as f:
|
|
json.dump(config_dict, f, indent=4)
|
|
|
|
# 将字典保存为json格式
|
|
def save_as_json(config_dict, output_file_path):
|
|
with open(output_file_path, 'w') as f:
|
|
json.dump(config_dict, f, indent=4)
|
|
|
|
# 主程序
|
|
if __name__ == "__main__":
|
|
input_file = 'Massage/aucpuncture2point/configs/body_2d_keypoint/rtmo/coco/rtmo-s_8xb32-600e_coco-640x640.py' # 这里填入你要转换的.py文件路径
|
|
output_file = 'Massage/aucpuncture2point/configs/body_2d_keypoint/rtmo/coco/rtmo-s_8xb32-600e_coco-640x640.json' # 输出的json文件路径
|
|
|
|
config_dict = read_py_config(input_file)
|
|
save_as_json(config_dict, output_file)
|
|
print(f"配置文件已转换并保存为 {output_file}")
|