PatchAugNet
PatchAugNet copied to clipboard
你好,感谢你们的开源工作,能否用自己的数据进行验证呢?还有就是想把生成的全局描述子提取出来保存,是这部分代码吗?
if save: # global desc g_desc_pkl = os.path.join(g_desc_dir, f"{beg_idx + b_i}.bin") with open(g_desc_pkl, 'wb') as handle: pickle.dump(g_desc, handle, protocol=pickle.HIGHEST_PROTOCOL) # local desc l_desc_pkl = os.path.join(l_desc_dir, f"{beg_idx + b_i}.bin") with open(l_desc_pkl, 'wb') as handle: pickle.dump((l_pos, l_desc, norm_meta), handle, protocol=pickle.HIGHEST_PROTOCOL)