update for bakeup

tracking/utils/drawtracks.py

@@ -98,11 +98,12 @@ def draw_all_trajectories(vts, edgeline, save_dir, file, draw5p=False):
     H, W = img.shape[:2]
     cv2.line(img, (int(W/2), 0), (int(W/2), H), (128, 255, 128), 2)
     
-    imgpth = save_dir.joinpath(f"{file}_show.png")
-    cv2.imwrite(str(imgpth), img)
+    
+    # imgpth = save_dir.joinpath(f"{file}_show.png")
+    # cv2.imwrite(str(imgpth), img)
 
     if not draw5p:
-        return
+        return img
         
     ''' tracks 5点轨迹'''
     trackpth = save_dir / Path("trajectory") / Path(f"{file}")

tracking/utils/read_data.py

@@ -151,7 +151,8 @@ def read_deletedBarcode_file(filePth):
         dict, barcode_list, similarity_list =  {}, [], []
   
         clean_lines = [line.strip().replace("'", '').replace('"', '') for line in lines]
-        for line in clean_lines:
+        
+        for i, line in enumerate(clean_lines):
             stripped_line = line.strip()
             if not stripped_line:
                 if len(barcode_list): dict['barcode'] = barcode_list
@@ -163,8 +164,16 @@ def read_deletedBarcode_file(filePth):
                 continue
             
             # print(line)
-            label = line.split(':')[0]
-            value = line.split(':')[1]
+            try:
+                label = line.split(':')[0]
+                value = line.split(':')[1]
+            except Exception as e:
+                print(f'Error: {e}')    
+            
+            
+            
+            
+            
             if label == 'SeqDir':
                 dict['SeqDir'] = value
             if label == 'Deleted':