Update track_detection.py

scripts/track_detection.py

@@ -70,7 +70,7 @@ def track_location(in_loc_dir, out_loc_dir, metadata_path, config, verbose):
 
             track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose)
 
-def track(in_loc_dir, out_loc_dir, metadata_path, seq_name, config, verbose):
+def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
 
     json_path = os.path.join(in_loc_dir, seq, 'pred.json')
     inference_path = os.path.join(in_loc_dir, seq, 'inference.pt')
@@ -98,7 +98,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq_name, config, verbose):
     with open(metadata_path, 'r') as f:
         json_object = json.loads(f.read())
         for seq in json_object:
-            if seq['clip_name'] == seq_name:
+            if seq['clip_name'] == seq:
                 image_meter_width = seq['x_meter_stop'] - seq['x_meter_start']
                 image_meter_height = seq['y_meter_stop'] - seq['y_meter_start']