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