squash me

angel_system/data/medical/load_bbn_data.py

@@ -262,7 +262,7 @@ def activity_label_fixes(task, activity_label, target):
     return label, label_id
 
 
-def bbn_activity_txt_to_csv(task, root_dir, output_dir):
+def bbn_activity_txt_to_csv(task, root_dir, output_dir, label_version):
     """
     Generate DIVE csv format activity annotations from BBN's text annotations
 
@@ -298,7 +298,7 @@ def bbn_activity_txt_to_csv(task, root_dir, output_dir):
             lines = action_f.readlines()
 
         # Create output csv
-        csv_fn = f"{output_dir}/{video_name}_activity_labels_v2.csv"
+        csv_fn = f"{output_dir}/{video_name}_activity_labels_v{label_version}.csv"
         csv_f = open(csv_fn, "w")
         csv_f.write(
             "# 1: Detection or Track-id,2: Video or Image Identifier,3: Unique Frame Identifier,4-7: Img-bbox(TL_x,TL_y,BR_x,BR_y),8: Detection or Length Confidence,9: Target Length (0 or -1 if invalid),10-11+: Repeated Species,Confidence Pairs or Attributes\n"