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

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())