def test_sub_blocks(): start = [8] quant = [[64,8,8]] color = ['hsv'] for s in start: for q in quant: print('--- # -- ') db_desc = SubBlockDescriptor(db) q1_desc = SubBlockDescriptor(qs1_w1) q2_desc = SubBlockDescriptor(qs2_w1,masks=True,mask_path=mask_root) db_desc.compute_descriptors(grid_blocks=[s,s],quantify=q,color_space=color[0]) q1_desc.compute_descriptors(grid_blocks=[s,s],quantify=q,color_space=color[0]) q2_desc.compute_descriptors(grid_blocks=[s,s],quantify=q,color_space=color[0]) # -- SEARCH -- # q1_search = Searcher(db_desc.result,q1_desc.result) q2_search = Searcher(db_desc.result,q2_desc.result) q1_desc.clear_memory() q2_desc.clear_memory() db_desc.clear_memory() q1_search.search(limit=3) q2_search.search(limit=3) # -- EVALUATION -- # q1_eval = EvaluateDescriptors(q1_search.result,res_root+os.sep+'gt_corresps1.pkl') q2_eval = EvaluateDescriptors(q2_search.result,res_root+os.sep+'gt_corresps2.pkl') q1_search.clear_memory() q2_search.clear_memory() q1_eval.compute_mapatk(limit=1) q2_eval.compute_mapatk(limit=1) filename = res_root+os.sep+'tests'+os.sep+'sub_res_'+str(s)+'_'+str(q[0])+'.pkl' with open(filename,'wb') as f: pickle.dump(q1_eval.score,f) pickle.dump(q2_eval.score,f) print('--- # -- ')
def test_level_desc(): level = [2] start = [5,6] jump = [2] quant = [[16,8,8],[24,12,12]] color = ['hsv'] for l in level: for s in start: for j in jump: for q in quant: print('--- # -- ') db_desc = LevelDescriptor(db) q1_desc = LevelDescriptor(qs1_w1) q2_desc = LevelDescriptor(qs2_w1,masks=True,mask_path=mask_root) db_desc.compute_descriptors(levels=l,init_quant=q,start=s,jump=j,color_space=color[0]) q1_desc.compute_descriptors(levels=l,init_quant=q,start=s,jump=j,color_space=color[0]) q2_desc.compute_descriptors(levels=l,init_quant=q,start=s,jump=j,color_space=color[0]) # -- SEARCH -- # q1_search = Searcher(db_desc.result,q1_desc.result) q2_search = Searcher(db_desc.result,q2_desc.result) db_desc.clear_memory() q1_desc.clear_memory() q2_desc.clear_memory() q1_search.search(limit=3) q2_search.search(limit=3) # -- EVALUATION -- # q1_eval = EvaluateDescriptors(q1_search.result,res_root+os.sep+'gt_corresps1.pkl') q2_eval = EvaluateDescriptors(q2_search.result,res_root+os.sep+'gt_corresps2.pkl') q1_search.clear_memory() q2_search.clear_memory() q1_eval.compute_mapatk(limit=1) q2_eval.compute_mapatk(limit=1) filename = res_root+os.sep+'tests'+os.sep+'lev_res_'+str(l)+'_'+str(s)+'_'+str(j)+'_'+str(q[0])+'.pkl' with open(filename,'wb') as f: pickle.dump(q1_eval.score,f) pickle.dump(q2_eval.score,f) print('--- # -- ')