diff --git a/omnidet/utils.py b/omnidet/utils.py index 155c7fc..65181ae 100644 --- a/omnidet/utils.py +++ b/omnidet/utils.py @@ -240,7 +240,7 @@ def __init__(self, num_classes, dataset, ignore_index=None): self.dataset = dataset self.classes = dict(woodscape_raw=["void", "road", "lanemarks", "curb", "person", - "rider", "vehicles", "bicycle", "motorcycle""traffic_sign"], + "rider", "vehicles", "bicycle", "motorcycle", "traffic_sign"], motion=['static', 'motion'], ) self.reset()