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