diff --git a/_modules/tracklab/wrappers/datasets/soccernet/soccernet_mot.html b/_modules/tracklab/wrappers/datasets/soccernet/soccernet_mot.html index 927720ff..947792dc 100644 --- a/_modules/tracklab/wrappers/datasets/soccernet/soccernet_mot.html +++ b/_modules/tracklab/wrappers/datasets/soccernet/soccernet_mot.html @@ -308,10 +308,10 @@

Source code for tracklab.wrappers.datasets.soccernet.soccernet_mot

image_metadata = pd.concat(image_metadata_list, ignore_index=True) detections = pd.concat(detections_list, ignore_index=True) - # Sort to obtain deterministic ids and index + # Use video_id, image_id, track_id as unique id detections = detections.sort_values(by=['video_id', 'image_id', 'track_id'], ascending=[True, True, True]) - detections = detections.reset_index(drop=True) - detections['id'] = detections.index + detections['id'] = detections['video_id'].astype(str) + "_" + detections['image_id'].astype(str) + "_" + detections[ + 'track_id'].astype(str) # Add category id to detections category_to_id = {category['name']: category['id'] for category in categories_list}