From e36154d642d733fd7621200553e53671c78b0fb7 Mon Sep 17 00:00:00 2001 From: nicolamarinello Date: Fri, 24 Mar 2023 17:35:01 +0100 Subject: [PATCH] Align with official BDD100K submission format --- docs/BDD100k-format.txt | 4 ++-- trackeval/datasets/bdd100k.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/BDD100k-format.txt b/docs/BDD100k-format.txt index 3939aa3b..888b5f28 100644 --- a/docs/BDD100k-format.txt +++ b/docs/BDD100k-format.txt @@ -28,12 +28,12 @@ Each json file contains a list of frame objects, and each frame object has the f - name: string - videoName: string -- index: int +- frameIndex: int - labels: [ ] - id: string - category: string - attributes: - - Crowd: boolean + - crowd: boolean - Occluded: boolean - Truncated: boolean - box2d: diff --git a/trackeval/datasets/bdd100k.py b/trackeval/datasets/bdd100k.py index cc4fd06e..8d8aa2c3 100644 --- a/trackeval/datasets/bdd100k.py +++ b/trackeval/datasets/bdd100k.py @@ -116,7 +116,7 @@ def _load_raw_file(self, tracker, seq, is_gt): data = json.load(f) # sort data by frame index - data = sorted(data, key=lambda x: x['index']) + data = sorted(data, key=lambda x: x['frameIndex']) # check sequence length if is_gt: @@ -139,7 +139,7 @@ def _load_raw_file(self, tracker, seq, is_gt): for i in range(len(data[t]['labels'])): ann = data[t]['labels'][i] if is_gt and (ann['category'] in self.distractor_classes or 'attributes' in ann.keys() - and ann['attributes']['Crowd']): + and ann['attributes']['crowd']): ig_ids.append(i) else: keep_ids.append(i)