modified for site test

tracking/goodmatch.py

@@ -30,6 +30,26 @@ from utils.drawtracks import plot_frameID_y2, draw_all_trajectories
 
 from utils.mergetrack import readDict
 
+import csv
+
+
+def read_csv_file():
+    
+    file_path = r'D:\DeepLearning\yolov5_track\tracking\matching\featdata\Similarity.csv'   
+    with open(file_path, mode='r', newline='') as file:    
+        data = list(csv.reader(file)) 
+    
+    matrix = []
+    for i in range(1, len(data)):
+        matrix.append(data[i][1:])
+        
+    matrix = np.array(matrix, dtype = np.float32)   
+    
+    simil = 1 + (matrix-1)/2
+
+    print("done!!!")
+
+
 
 
 def get_img_filename(imgpath = r'./matching/images/' ):
@@ -747,7 +767,7 @@ def main():
     # imgsample_cleaning()
     
     '''3.1 计算事件间相似度: 将 front、back 的所有 track 特征合并'''
-    # calculate_similarity()
+    calculate_similarity()
     
     '''3.2 计算事件间相似度: 考虑前后摄的不同组合，或 track 间的不同组合'''
     # calculate_similarity_track()
@@ -766,8 +786,29 @@ def main():
     
                
 if __name__ == "__main__":
-    save_dir = Path(f'./result/') 
+    # save_dir = Path(f'./result/') 
+    # read_csv_file()
+    
     main()
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+    
+