把xview_train和xview_val对应的键值对删除

最新推荐文章于 2024-07-30 09:11:33 发布

鱼儿会飞吗

最新推荐文章于 2024-07-30 09:11:33 发布

阅读量142

点赞数 5

文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/qq_34425255/article/details/140787348

版权

import pickle

pkl_file_path = '/root/pyskl/tools/data/nturgbd/ntu60_hrnet.pkl'
with open(pkl_file_path, 'rb') as file:
    pkl_data = pickle.load(file)

if "xview_train" in pkl_data["split"]:
    del pkl_data["split"]["xview_train"]
if "xview_val" in pkl_data["split"]:
    del pkl_data["split"]["xview_val"]

new_pkl_file_path = '/root/pyskl/tools/data/ntu60_hrnet_modified.pkl'
with open(new_pkl_file_path, 'wb') as file:
    pickle.dump(pkl_data, file)

pyskl/tools/data/README.md at main · kennymckormick/pyskl · GitHub