python 保存自定义类对象到本地并读取

HelloJinYe

于 2020-08-20 20:41:46 发布

阅读量4.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： python&conda 文章标签： python

本文链接：https://blog.csdn.net/HelloJinYe/article/details/108133178

python&conda 专栏收录该内容

10 篇文章

订阅专栏

本文详细介绍了如何使用Python创建KDTree，并将其保存到文件中以便后续读取和使用，有效减少重复创建树的时间成本。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.保存对象

# 创建kdtree
my_tree = MyKDTree(npz_dir, 3, choose_keypts_num)
my_tree.create()
time_end = time.time()
print('kdtree create time cost ', time_end - time_start, ' s')

# 保存kdtree
out_put = open("my_tree.pkl", 'wb')
tree_str = pickle.dumps(my_tree)
out_put.write(tree_str)
out_put.close()

2.读取对象

# 读取kdtree 缩短时间
my_tree = MyKDTree()
with open("my_tree.pkl", 'rb') as file:
    my_tree = pickle.loads(file.read())