Update test_st_market.py

Wanggcong · Dec 14, 2018 · 667a0c7 · 667a0c7
1 parent 0de0668
commit 667a0c7
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/test_st_market.py b/test_st_market.py
@@ -163,12 +163,6 @@ def get_id(img_path):
 gallery_cam,gallery_label, gallery_frames = get_id(gallery_path)
 query_cam,query_label, query_frames = get_id(query_path)
 
-# gallery_path = image_datasets['gallery'].imgs
-# query_path = image_datasets['query'].imgs
-
-# gallery_cam,gallery_label = get_id(gallery_path)
-# query_cam,query_label = get_id(query_path)
-
 ######################################################################
 # Load Collected data Trained model
 class_num=751
@@ -199,10 +193,6 @@ def get_id(img_path):
 gallery_feature = extract_feature(model,dataloaders['gallery'])
 query_feature = extract_feature(model,dataloaders['query'])
 
-# # Save to Matlab for check
-# result = {'gallery_f':gallery_feature.numpy(),'gallery_label':gallery_label,'gallery_cam':gallery_cam,'query_f':query_feature.numpy(),'query_label':query_label,'query_cam':query_cam}
-# scipy.io.savemat('pytorch_result.mat',result)
-
 # Save to Matlab for check
 result = {'gallery_f':gallery_feature.numpy(),'gallery_label':gallery_label,'gallery_cam':gallery_cam,'gallery_frames':gallery_frames,'query_f':query_feature.numpy(),'query_label':query_label,'query_cam':query_cam,'query_frames':query_frames}
 scipy.io.savemat('model/'+name+'/'+'pytorch_result.mat',result)