Update bdd100k.py

haofengac · Mar 11, 2019 · 2773e4a · 2773e4a
1 parent 2c5de35
commit 2773e4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/maskrcnn_benchmark/data/datasets/bdd100k.py b/maskrcnn_benchmark/data/datasets/bdd100k.py
@@ -81,7 +81,7 @@ def __getitem__(self, idx):
         classes = []
         for label in annotations['labels']:
             # TODO: further filter annotations if needed
-
+            classes += [label['category']]
             boxes += [
                 label['box2d']['x1'],
                 label['box2d']['y1'],